python 第一行非零_python – 在Pandas DataFrame的每一行中查找第一个非零值

我有一个Pandas DataFrame：

import pandas as pd

df = pd.DataFrame([[0.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0],

[1.0, 0.0, 1.0, 3.0, 1.0, 1.0, 7.0, 0.0],

[0.0, 0.0, 13.0, 14.0, 15.0, 16.0, 17.0, 18.0]

]

, columns=['A', 'B', 'C', 'D', 'E', 'F', 'G', 'H'])

A B C D E F G H

0 0.0 2.0 3.0 4.0 5.0 6.0 7.0 8.0

1 1.0 0.0 1.0 3.0 1.0 1.0 7.0 0.0

2 0.0 0.0 13.0 14.0 15.0 16.0 17.0 18.0

我想返回每行中第一个非零值的序列(不是列表).这当前有效,但查找返回一个列表而不是一个系列(我知道我可以将列表转换为一个系列)但我假设有一个更好的方法：

first_nonzero_colnames = (df > 0).idxmax(axis=1, skipna=True)

df.lookup(first_nonzero_colnames.index, first_nonzero_colnames.values)

[ 2. 1. 13.]

我可以使用.apply但我想避免它.