我已将.csv文件的内容放在数据列表中在Jupyter笔记本中使用以下代码
data = []
with open("president_county_candidate.csv", "r") as f:
contents = csv.reader(f)
for c in contents:
data.append(c)
我只能通过索引号获得一个元素,但这会给我列表的整行。如何选择特定元素和计数?在图像中,您可以看到列表(数据(的内容。
数据
您可以使用pandas
库以数据帧的形式读取csv,也可以对其执行各种操作。参考以下代码:
import pandas as pd
df = pd.read_csv('president_county_candidate.csv')
print(df.shape)
这将为您提供CSV文件中的行数和列数。
此外,为了提取任何特定的列,您可以使用:
newdf = df[['candidates', 'votes']]
这将为您提供具有上述列的新数据帧。
此外,请为您在问题中提到的方法找到解决方案
您可以提取列索引,然后在解析内容时,提及所需列的索引号。
例如:
data = []
with open("president_county_candidate.csv", "r") as f:
contents = csv.reader(f)
for c in contents:
data.append([c[2], c[4]])
这将只获得候选人和投票的数据。
注意:最好使用list.index('columnName')
获取特定列的索引号,并传递相同的变量,而不是硬编码索引。