在 Pandas DataFrame 中选择多列

要在 Pandas DataFrame 中选择多个列,我们可以从现有的 DataFrame 创建一个新的 DataFrame

步骤

  • 创建二维、大小可变、潜在异构的表格数据df

  • 打印输入数据帧。

  • 创建一个新的 DataFrame df1,并选择多列。

  • 打印具有多个选定列的新 DataFrame。

示例

import pandas as pd

df = pd.DataFrame(
   {
      "x": [5, 2, 1, 9],
      "y": [4, 1, 5, 10],
      "z": [4, 1, 5, 0]
   }
)

print "Input DataFrame is:\n", df

df1 = df[['x', 'y']]
print "After selecting multiple columns:\n", df1
输出结果
Input DataFrame is:
   x  y  z
0  5  4  4
1  2  1  1
2  1  5  5
3  9 10  0

After selecting multiple columns:
   x  y
0  5  4
1  2  1
2  1  5
3  9 10