Python CSV 或 Pd 数据框解析问题

Question

我有一个 csv 文件或 pd 数据框对象，在两个轴（列和行）上都有标签。我需要检查每个单元格的值，如果值为 True，我需要为这些值打印相应的轴标签（两个轴）。我花了一些时间在 Python 中找不到解决方案。非常感谢任何帮助。

类似这样的：

['', 'column_name_1', 'column_name_2', 'column_name_3',...] 
['Row_1_name', 'False', 'False', 'True'...] 
['Row_2_name', 'False', 'False', 'True'...]

感谢您的帮助。

Answer 1

这是一个简单的示例，它通过对所有列应用打印函数来打印数据框中所有 True 值的索引和列：

import pandas
import io

data = '''col1,col2
label1,False,True
label2,True,False
label3,False,True'''

df = pd.read_csv(io.StringIO(data), engine='python')

def print_labels(x):
    for label, val in x.iteritems():
        if val: #insert your own validation here (or filter the column before the loop)
            print(label, x.name)

df.apply(print_labels, axis=0)

输出：

label2 col1
label1 col2
label3 col2