如果行值=1,可以通过以下步骤从Dataframe生成列名列表:
以下是一个示例代码:
import pandas as pd
# 假设df是一个Dataframe对象
# 获取列名列表
column_names = df.columns.tolist()
# 创建空列表来存储行值为1的列名
column_list = []
# 遍历列名列表
for column_name, column_values in df.iteritems():
# 检查行值是否为1
if 1 in column_values.values:
# 将列名添加到column_list中
column_list.append(column_name)
# 返回行值为1的列名列表
print(column_list)
请注意,以上代码仅适用于行值为1的情况。如果行值可能是其他值,需要根据具体需求进行相应的修改。
领取专属 10元无门槛券
手把手带您无忧上云