dataframe保留指定列

要保留DataFrame中的指定列，可以使用`loc`或`iloc`方法。

`loc`方法根据列标签进行索引，语法如下：

new_df = df.loc[:, ['列1', '列2', ...]]

其中，`:`表示保留所有行，`['列1', '列2', ...]`表示要保留的列标签。

`iloc`方法根据列位置进行索引，语法如下：

new_df = df.iloc[:, [列1的位置, 列2的位置, ...]]

其中，`:`表示保留所有行，`[列1的位置, 列2的位置, ...]`表示要保留的列位置。

例如，我们有一个DataFrame `df`：

import pandas as pd

data = {'姓名': ['张三', '李四', '王五'], '年龄': [20, 25, 30], '性别': ['男', '女', '男']}
df = pd.DataFrame(data)
print(df)

输出：

   姓名  年龄 性别
0  张三  20  男
1  李四  25  女
2  王五  30  男

如果要保留`姓名`和`年龄`两列，可以使用`loc`或`iloc`方法：

new_df = df.loc[:, ['姓名', '年龄']]
# 或者
new_df = df.iloc[:, [0, 1]]
print(new_df)

输出：

   姓名  年龄
0  张三  20
1  李四  25
2  王五  30