python保存csv指定列

要保存CSV文件中的指定列，可以使用pandas库。以下是一个示例代码：

import pandas as pd

# 读取CSV文件
df = pd.read_csv('data.csv')

# 选取指定列
selected_columns = ['column1', 'column2']  # 替换为你需要的列名
df_selected = df[selected_columns]

# 保存为新的CSV文件
df_selected.to_csv('selected_columns.csv', index=False)

在这个示例中，我们假设有一个名为"data.csv"的CSV文件，然后我们从中选择"column1"和"column2"这两列，并将结果保存为"selected_columns.csv"文件。你可以根据需要修改列名和文件名。

相关问题

python保存csv指定列的数据

要保存 CSV 文件中指定列的数据，你可以使用 Python 的 `csv` 模块来处理。下面是一个示例代码：

import csv

def save_selected_columns(input_file, output_file, selected_columns):
    with open(input_file, 'r', newline='') as file:
        reader = csv.reader(file)
        headers = next(reader)  # 读取文件的第一行作为表头

        # 确定所选列的索引
        selected_indexes = [headers.index(col) for col in selected_columns]

        with open(output_file, 'w', newline='') as outfile:
            writer = csv.writer(outfile)
            writer.writerow(selected_columns)  # 写入所选列的表头

            for row in reader:
                selected_row = [row[i] for i in selected_indexes]
                writer.writerow(selected_row)  # 写入所选列的数据

# 示例用法
input_file = 'input.csv'
output_file = 'output.csv'
selected_columns = ['列名1', '列名2', '列名3']

save_selected_columns(input_file, output_file, selected_columns)

在上述代码中，`input_file` 是输入的 CSV 文件路径，`output_file` 是输出的 CSV 文件路径，`selected_columns` 是你想要保存的列的列表。请将代码中的 `'列名1', '列名2', '列名3'` 替换为你要保存的实际列名。

运行该代码后，会读取 `input.csv` 文件中的数据，提取指定的列，并将结果保存到 `output.csv` 文件中。

python保存csv指定列的数据 具体例子

当你想要保存CSV文件中的指定列数据时，你可以使用Python的pandas库来实现。以下是一个具体的例子：

import pandas as pd

# 读取CSV文件
df = pd.read_csv('example.csv')

# 选择指定列
selected_columns = ['column1', 'column2']  # 选择要保存的列名
df_selected = df[selected_columns]

# 保存为新的CSV文件
df_selected.to_csv('selected_columns.csv', index=False)

在上面的例子中，我们首先使用`pd.read_csv()`函数读取名为`example.csv`的CSV文件，并将数据存储在一个名为`df`的DataFrame对象中。

然后，我们定义一个`selected_columns`列表，其中包含你想要保存的列名。

接下来，我们通过从原始DataFrame中选择指定列创建一个新的DataFrame `df_selected`。

最后，我们使用`to_csv()`函数将新的DataFrame保存为名为`selected_columns.csv`的CSV文件。`index=False`参数表示不保存行索引。

你可以根据需要修改列名和文件名。希望对你有所帮助！