如何在熊猫中过滤一些列值并使用这些值生成新列？

Question

sht.head()

如何创建包含A：J中所有非零值的列K？

预期结果将是：K = ["A001", "B001", "C001", "D001", "E001", "A002", ...]

Answer 1

在agg中使用axis=1

df['K'] = df.agg(lambda s: tuple(s[s!=0]), axis=1)

Answer 2

尝试一下

sht['K']=pd.Series((sht[sht!=0]).values.flatten()).dropna()