在Pandas中查找第一个跨列的非零条目

import pandas as pd

# 创建一个示例DataFrame
data = {'A': [0, 0, 0, 1, 0],
        'B': [0, 2, 0, 0, 0],
        'C': [3, 0, 0, 0, 0],
        'D': [0, 0, 0, 0, 4]}
df = pd.DataFrame(data)

# 使用idxmax()函数查找第一个跨列的非零条目
result = df.apply(lambda x: x.idxmax(), axis=1)

print(result)

0    C
1    B
2    A
3    A
4    D
dtype: object