python读取csv某一列存入数组_python 读取.csv文件数据到数组(矩阵)的实例讲解
时间: 2023-08-13 14:05:02 浏览: 399
好的,下面是一个Python读取csv文件数据到数组(矩阵)的例子:
假设我们有一个名为data.csv的csv文件,其中包含以下内容:
```csv
Name,Age,Country
John,25,USA
Alice,30,Canada
Bob,20,UK
```
现在我们想要读取csv文件中的Age列,并将其存储到一个数组中。
首先,我们需要使用Python内置的csv模块来打开csv文件并读取其内容。代码如下:
```python
import csv
with open('data.csv', 'r') as file:
reader = csv.reader(file)
for row in reader:
print(row)
```
上面的代码将打开data.csv文件并将其内容逐行打印出来。现在我们需要找到Age列并将其存储到一个数组中。为此,我们可以使用以下代码:
```python
import csv
ages = []
with open('data.csv', 'r') as file:
reader = csv.reader(file)
for row in reader:
age = row[1] # 获取Age列
ages.append(age)
print(ages) # 打印Age列数组
```
在上面的代码中,我们创建了一个空的数组ages,然后使用csv.reader逐行读取csv文件中的数据。对于每一行数据,我们使用row[1]获取Age列的值,并将其添加到数组ages中。最后,我们打印出ages数组。
希望这个例子能够帮助你读取csv文件中的数据并将其存储到数组中。
阅读全文