我有 3 列，其值為 A、B 或 C 我想比較這 3 列并給出其中值計(jì)數(shù)超過(guò) 1 的輸出。如果計(jì)數(shù)相等則輸出將為“-”輸入： | col1 | col2 | col3 | |-------|-------|-------| | A | A | B | | A | B | B | | C | B | C | | A | B | C |輸出： | col1 | col2 | col3 | Result| |-------|-------|-------|-------| | A | A | B | A | | A | B | B | B | | C | B | C | C | | A | B | C | - |
2 回答
烙印99
TA貢獻(xiàn)1829條經(jīng)驗(yàn) 獲得超13個(gè)贊
讓我們嘗試Counter獲取最常見的元素：
from collections import Counter
def most_common():
for s in df.to_numpy():
k, v = Counter(s).most_common(1)[0]
yield '-' if v == 1 else k
df['Result'] = list(most_common())
col1 col2 col3 Result
0 A A B A
1 A B B B
2 C B C C
3 A B C -
慕俠2389804
TA貢獻(xiàn)1719條經(jīng)驗(yàn) 獲得超6個(gè)贊
如果性能不是一個(gè)大問(wèn)題，您可以編寫一個(gè)按行方法返回出現(xiàn)多次的值：
def appear_more_than_once(s):
cnt = s.value_counts()
if len(cnt) > 0 and cnt.iloc[0] > 1:
return cnt.index[0]
else:
return '-'
df['Result'] = df.apply(appear_more_than_once, axis=1)
print(df)
# col1 col2 col3 Result
#0 A A B A
#1 A B B B
#2 A B C -
https://uscript.co/public/Google_108617488638529745626/python/4e5d36bc.py
添加回答
舉報(bào)
0/150
提交
取消