如何编写函数以在熊猫中提取 n+1 列答案

【问题标题】：How to write function to extract n+1 column in pandas如何编写函数以在熊猫中提取 n+1 列
【发布时间】：2019-12-02 19:05:58
【问题描述】：

我有一个包含 200 列的 excel 文件。第一列为否。访问次数，其他列是该访问次数的人数数据

我想编写一个函数，以便我有多个带有访问列和 A 的数据框；访问列和B，等等（我想写一个函数，因为将来列的数量会增加，我想自动化这个过程）。另外，我想用0去掉数据。

期望的输出：

数据框 1：

数据框 2：

这是我的第一个问题。很抱歉，如果它没有正确构图。感谢您的帮助。

【问题讨论】：

【解决方案1】：

使用DataFrame.items:

for i,col in df.set_index('visits').items():
    print(col[col.ne(0)].to_frame(i).reset_index())

您可以创建一个字典以按列名称保存

dfs={i:col[col.ne(0)].to_frame(i).reset_index() for i,col in df.set_index('visits').items()}

【讨论】：