python读取文件,将几列的数据提取出来保存在另一个文件

可以使用Python的pandas库来读取文件并提取数据。以下是一个示例代码：

import pandas as pd

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

# 提取需要的列
new_df = df[['column1', 'column2', 'column3']]

# 保存到另一个文件
new_df.to_csv('output_file.csv', index=False)

其中，`input_file.csv`是需要读取的文件名，`column1`、`column2`和`column3`是需要提取的列名，`output_file.csv`是保存结果的文件名。这段代码会将`input_file.csv`中的`column1`、`column2`和`column3`三列数据提取出来，并保存到`output_file.csv`中。

相关问题

python读取excel数据几列

在Python中，你可以使用`pandas`库来读取Excel数据。要读取Excel中的几列数据，可以使用`pandas`库的`read_excel`函数并指定需要读取的列。

下面是一个示例代码，展示如何读取Excel中的两列数据：

import pandas as pd

# 读取Excel文件
data = pd.read_excel('your_file.xlsx')

# 选择需要读取的列
selected_columns = ['column1', 'column2']

# 提取所选列的数据
selected_data = data[selected_columns]

# 打印结果
print(selected_data)

在上面的示例中，你需要将`your_file.xlsx`替换为你要读取的Excel文件的路径。`column1`和`column2`是你想要读取的列的名称，你可以根据实际情况进行更改。最后，通过打印`selected_data`，你将得到所选列的数据。

python提取csv文件几列并保存为csv

要提取csv文件的几列并保存为csv，可以使用以下代码：

import csv

# 要提取的列的索引
columns = [0, 3, 5]

# 读取原始csv文件
with open('原始文件.csv', 'r') as f:
    reader = csv.reader(f)
    header = next(reader)  # 获取表头
    
    # 获取要提取的列的表头
    header_new = [header[i] for i in columns]
    
    # 逐行读取数据并提取要保留的列
    rows_new = []
    for row in reader:
        row_new = [row[i] for i in columns]
        rows_new.append(row_new)
        
# 保存提取后的csv文件
with open('提取后的文件.csv', 'w', newline='') as f:
    writer = csv.writer(f)
    writer.writerow(header_new)
    writer.writerows(rows_new)

在这个代码中，首先定义了要提取的列的索引，然后使用csv模块读取原始csv文件，获取表头和数据。接着，根据要提取的列的索引，获取要保留的列的表头和数据，并保存到一个新的列表中。最后，使用csv模块将新的表头和数据写入到新的csv文件中。