以下是一个示例代码,演示如何按照列的编号选择:
import pandas as pd
# 创建一个示例DataFrame
data = {'A': [1, 2, 3],
'B': [4, 5, 6],
'C': [7, 8, 9]}
df = pd.DataFrame(data)
# 按照列的编号选择
selected_columns = df.iloc[:, [0, 2]] # 选择第一列和第三列
print(selected_columns)
输出结果为:
A C
0 1 7
1 2 8
2 3 9
在示例代码中,首先创建了一个示例的DataFrame。然后,使用iloc
方法按照列的编号选择需要的列。在这个例子中,选择了第一列和第三列,所以使用了[:, [0, 2]]
这个索引方式。最后,打印出选择的列的结果。
上一篇:按照列的编号排序