下面是一个示例代码,展示了如何使用itertuples方法来按列名选择索引。
import pandas as pd
# 创建示例数据
data = {'A': [1, 2, 3, 4, 5],
'B': [6, 7, 8, 9, 10],
'C': [11, 12, 13, 14, 15]}
df = pd.DataFrame(data)
# 使用itertuples方法按列名选择索引
selected_rows = []
for row in df.itertuples(index=True):
if row.A > 2: # 选择A列大于2的行
selected_rows.append(row.Index)
print(selected_rows)
输出结果为:[2, 3, 4],表示选择了A列大于2的行的索引。
上一篇:按列名替换列的值
下一篇:按列名在Athena中解析CSV