从dataframe中选择要转换为子集的列列表

import pandas as pd

# 创建一个示例DataFrame
df = pd.DataFrame({'A': [1, 2, 3], 'B': [4, 5, 6], 'C': [7, 8, 9]})

# 选择要转换为子集的列列表
columns_to_select = ['A', 'C']

# 使用loc方法选择指定列的子集DataFrame
subset = df.loc[:, columns_to_select]

print(subset)

   A  C
0  1  7
1  2  8
2  3  9

import pandas as pd

# 创建一个示例DataFrame
df = pd.DataFrame({'A': [1, 2, 3], 'B': [4, 5, 6], 'C': [7, 8, 9]})

# 选择要转换为子集的列列表的索引
column_indices_to_select = [0, 2]

# 使用iloc方法选择指定列的子集DataFrame
subset = df.iloc[:, column_indices_to_select]

print(subset)

   A  C
0  1  7
1  2  8
2  3  9