python-从DataFrame中按行提取列名称到Series

内容导读

互联网集市收集整理的这篇技术教程文章主要介绍了python-从DataFrame中按行提取列名称到Series，小编现在分享给大家，供广大互联网技能从业者学习和参考。文章包含770字，纯文字阅读大概需要2分钟。

内容图文

我想将列表中的列名提取到按每行值过滤的Series

In [1]: import pandas as pd   

In [2]: df =pd.DataFrame({'colA':[1,0,1], 'colB':[0,0,1], 'colC':[1,0,0]})    

In [3]: print(df)

   colA  colB  colC
0     1     0     1
1     0     0     0
2     1     1     0

生成的系列应如下所示：

0    [colA, colC]
1              []
2    [colA, colB]
dtype: object

这是我想出的折磨解决方案：

In [4]: df2 = df.T

In [5]: l = [df2[df2[i]>0].index.values.tolist() for i in range(3)]

In [6]: print(pd.Series(l))

0    [colA, colC]
1              []
2    [colA, colB]
dtype: object

有没有那么折磨的方式呢？

解决方法:

您可以使用np.where假设您的数据帧由0和1组成,并根据结果创建一个Series：

x = np.where(df,df.columns,'')
pd.Series([' '.join(i).split() for i in x])
0    [colA, colC]
1              []
2    [colA, colB]

内容总结

以上是互联网集市为您收集整理的python-从DataFrame中按行提取列名称到Series全部内容，希望文章能够帮你解决python-从DataFrame中按行提取列名称到Series所遇到的程序开发问题。 如果觉得互联网集市技术教程内容还不错，欢迎将互联网集市网站推荐给程序员好友。

内容备注

版权声明：本文内容由互联网用户自发贡献，该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容， 请发送邮件至 gblab@vip.qq.com 举报，一经查实，本站将立刻删除。

内容手机端