Python读取CSV文件特定某两列的内容

如题，有的时候我们需要对csv或者Excel文件进行处理（Excel也可以另存为CSV文件再操作），一个包含这几千行，几十列的一个表，看着就很头大。但是这个大的一个表中，我们感兴趣的只是那么几列数据。

举个栗子，目标是拿到wordcupmatches.csv文件里面Home team goals和Away team goals两列的数据

__author__ = "lingjun"
# E-mail: [email protected]
# 微信公众号：小白CV

import csv

file = "D:/小象作业/ml-statistics-quartile/data/WorldCupMatches.csv"

with open(file,'r', encoding = 'UTF-8') as f :
    # TODO
    # 使用csv.DictReader读取文件中的信息
    reader = csv.DictReader(f)
   
    home_team_goals = []
    away_team_goals = []
    for row in reader :
        # TODO 
        # 将 'Home Team Goals' 、'Away Team Goals'中的每个元素以整型数据分别添加在相应的列表中
        home_team_goals.append(row['Home Team Goals'])
        away_team_goals.append(row['Away Team Goals'])
        
        #转变数据内形式，str2int
        #home_team_goals.append(int(row['Home Team Goals']))
        #away_team_goals.append(int(row['Away Team Goals']))
        
        
        
print(home_team_goals[:10])
print(away_team_goals[:10])

小白CV：公众号旨在专注CV（计算机视觉）、AI（人工智能）领域相关技术，文章内容主要围绕C++、Python编程技术，机器学习（ML）、深度学习（DL）、OpenCV等图像处理技术，深度发掘技术要点，记录学习工作中常用的操作，做你学习工作的问题小助手。只关注技术，做CV领域专业的知识分享平台。