如何在整个数据帧中找到每列中的最后一个非零元素？

Question

如何才能在数据帧的每一列中找到最后出现的非零元素？

输入

    A  B
0   0  1
1   0  2
2   9  0
3  10  0
4   0  0
5   0  0

产量

    A  B
0  10  2

Answer 1

您可以将0转换为缺失值，使用正向填充并通过索引选择最后一行，最后转换为整数：

df = df.mask(df==0).ffill().iloc[[-1]].astype(int)
print (df)
    A  B
5  10  2

Answer 2

这是使用ndarray.argmax和高级索引的一种方法：

first_max = df.values[df.ne(0).values.argmax(0), range(df.shape[1])]
out = pd.DataFrame([first_max], columns=df.columns)

---

df = pd.DataFrame({'A': [0,0,0,10,0,0] , 'B': [0,2,0,0,0,0]})

first_max = df.values[df.ne(0).values.argmax(0), range(df.shape[1])]
# array([10,  2])
pd.DataFrame([first_max], columns=df.columns)

    A  B
0  10  2

---

更新

为了找到最后一个非零：

row_ix = df.shape[0]-df.ne(0).values[::-1].argmax(0)-1
first_max = df.values[row_ix, range(df.shape[1])]
out = pd.DataFrame([first_max], columns=df.columns)

Answer 3

就像是：

results = {}
for column in df.columns:
    results[column] = df.loc[df[column]!=0, column].iloc[-1]

这将生成一个包含所有列作为键的字典，并将非零值作为值。

编辑：如果你想在数据帧中，加上单行的字典理解：

results = pd.DataFrame({column:[df.loc[df[column]!=0, column].iloc[-1]] for column in df.columns})

Answer 4

循环遍历列然后行，并存储最后一个非零变量

list = []* number_of_columns
for i in range(len(df)):
    dfcolumn = df[:,i]
    for item in dfcolumn:
        if item !=  0:
            list[i] = [i, item]

print(list)

Answer 5

使用itertools.dropwhile

特定

import itertools as it

import pandas as pd


df = pd.DataFrame(
    {"A": [0, 0, 9, 10, 0, 0], 
     "B": [1, 2, 0, 0, 0, 0]}
)

码

#3                 2                 1 
[next(it.dropwhile(lambda x: x == 0, reversed(col))) for _, col in df.iteritems()]

产量

[10, 2]

---

细节

对于DataFrame中的每一列，我们都希望如此

1. 反向迭代列，例如[0, 0, 10, 9, 0, 0]
2. 将所有零都删除到第一个非零元素，例如[10, 9, 0, 0]
3. 从迭代器中获取下一个元素，例如10