如何使用python访问csv文件中的每一列

Posted 2023-04-18

【中文标题】如何使用python访问csv文件中的每一列

【英文标题】：how to access each column in csv file using python

【发布时间】：2017-06-27 03:37:45

【问题描述】：

我有 .csv 数据集，它有两列，每一列都有节点，这个文件的一行代表两个节点之间的关系。 我想用它来使用 Python 的 networkx 库创建一个图形，但我不知道如何到达一行中每一列的每个单元格。这是我的代码。

import networkx as nx
import csv
G = nx.Graph()
with open('/pathToTheFile/edges.csv', newline='') as f:
    reader = csv.reader(f)
    counter = 0
    for row in reader:
        G.add_edge()#I will pass two nodes here to add an edge between them
        counter += 1
        if counter >= 65535:
            break

【参考方案1】：

当您遍历 csv 阅读器时，您会将文件的每一行作为一个列表。因此，您可以使用索引来访问一行中的第 0 个和第 1 个单元格，并在它们之间创建一条边。

import networkx as nx
import csv
G = nx.Graph()
with open('csv-networkx.csv', newline='') as f:
    reader = csv.reader(f)
    counter = 0
    for row in reader: # row is a list of the cells in this row.
        G.add_edge(row[0], row[1])
        counter += 1
        if counter >= 65535:
            break

【参考方案2】：

因为是.csv，所以可以用逗号分割：

values = row.split(',')
# 1st column value in row:
values[0]
# 2nd column value in row
values[1]