我有 3 列，其值為 A、B 或 C 我想比較這 3 列并給出其中值計數(shù)超過 1 的輸出。如果計數(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貢獻1829條經(jīng)驗 獲得超13個贊
讓我們嘗試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貢獻1719條經(jīng)驗 獲得超6個贊
如果性能不是一個大問題，您可以編寫一個按行方法返回出現(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
添加回答
舉報
0/150
提交
取消