首页 > 解决方案 > 如何使用通过熊猫从 csv 读取的值填充数组?

问题描述

我通过读取 csv 文件创建了一个DataFrame使用。pandas我想要做的是将行(对于第 1 列中的值)迭代到某个数组中,并对第 2 列中的值执行相同的操作以获取不同的数组。这似乎通常是一件相当容易的事情,所以我想我错过了一些东西,但是我在网上找不到太多不太复杂并且似乎没有做我想做的事。像这样 的堆栈问题似乎在问同样的事情,但答案又长又复杂。有没有办法在几行代码中做到这一点?这是我设置的:

import pandas as pd 

#available possible players
playerNames = []

df = pd.read_csv('Fantasy Week 1.csv')

我预计我应该能够做的是:

for row in df.columns[1]:
    playerNames.append(row)

但是,这不会返回所需的结果。

本质上,如果 df =
[1,2,3
4,5,6
7,8,9],我希望我的数组是 [1,4,7]

标签: pythonpandascsvdataframe

解决方案


做:

for row in df[df.columns[1]]:
    playerNames.append(row)

甚至更好:

print(df[df.columns[1]].tolist())

在这种情况下,您需要第一列的值:

for row in df[df.columns[0]]:
    playerNames.append(row)

甚至更好:

print(df[df.columns[0]].tolist())

推荐阅读