python只保留数据的某几列

如果想在 Python 中只保留数据的某几列，可以使用 pandas 库来处理数据。以下是一个示例代码，演示如何选择要保留的列：

import pandas as pd

# 创建一个示例数据集
data = {'列1': [1, 2, 3],
        '列2': [4, 5, 6],
        '列3': [7, 8, 9]}

# 将数据转换为 DataFrame
df = pd.DataFrame(data)

# 选择要保留的列
columns_to_keep = ['列1', '列3']
new_df = df[columns_to_keep]

# 打印结果
print(new_df)

输出结果将会是：

   列1  列3
0   1   7
1   2   8
2   3   9

在上面的示例中，使用 pandas 创建了一个 DataFrame，然后使用列的名称列表 columns_to_keep 选择要保留的列，并将结果存储在 new_df 中。

注意：在实际的应用中，可能会从文件或其他数据源中读取数据，并使用 pandas 进行处理。以上示例仅仅是一个简单的演示，可以根据自己的实际需求进行相应的调整。