当前位置:首页python > 正文

python对csv数据提取某几列

作者:野牛程序员:2023-07-11 19:32:44python阅读 3090

要在Python中从CSV文件中提取特定的列,你可以使用CSV模块或者Pandas库。下面我将为你提供两种方法的示例代码:

方法一:使用CSV模块

import csv

# 打开CSV文件
with open('data.csv', 'r') as file:
    reader = csv.reader(file)
    
    # 提取的列索引
    column_indices = [0, 2, 4]  # 假设你要提取第1、第3和第5列
    
    # 逐行读取CSV数据,并提取指定的列
    for row in reader:
        selected_columns = [row[i] for i in column_indices]
        print(selected_columns)

方法二:使用Pandas库

import pandas as pd

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

# 提取指定的列
selected_columns = df.iloc[:, [0, 2, 4]]  # 假设你要提取第1、第3和第5列

# 打印提取的列
print(selected_columns)

以上代码中,你需要将文件名 'data.csv' 替换为你实际使用的CSV文件名。在示例代码中,我们假设要提取的列的索引是 [0, 2, 4],你可以根据自己的需求修改为所需的列索引。

野牛程序员教少儿编程与信息学奥赛-微信|电话:15892516892
野牛程序员教少儿编程与信息学竞赛-微信|电话:15892516892
相关推荐

最新推荐

热门点击