如何获取 pandas dataframe 列的第一个非零值的索引

Question

我正在尝试获取 pandas dataframe 的特定列的第一个非零值的索引。

import pandas as pd

df = pd.DataFrame(
    [[0,0,0],
    [0,0,3],
    [1,5,7],
    [8,7,3]],
    columns=['c1','c2','c3'])

print(df)

Output：

    c1  c2  c3
0   0   0   0
1   0   0   3
2   1   5   7
3   8   7   3

我想在 c2 列中获得值为 5（第一个非零值）的索引 position。

Answer 1

您可以将boolean 索引与first_valid_index结合使用：

df.loc[df['c2'].ne(0), 'c2'].first_valid_index()

output： 2

注意。 如果列中只有零，则 output 将为None 。