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

Question

我有一个包含两列的 .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

Answer 1

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

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

Answer 2

当您遍历 csv reader 时，您会得到文件的每一行作为一个列表。因此，您可以使用索引访问一行中的第 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

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

how to access each column in csv file using python

python

graph

dataset

bigdata

data-science