python如何选取csv文件指定的列
时间: 2023-11-23 21:07:49 浏览: 239
要在Python中选择CSV文件中的指定列,你可以使用csv模块来读取和处理CSV文件。以下是几种方法可以实现这个目标:
方法一:使用csv.DictReader
你可以使用csv.DictReader来读取CSV文件,并通过指定列的标签来获取对应的数值。首先,你需要打开CSV文件,然后创建一个DictReader对象。接下来,你可以使用列表推导式来获取指定列的数值。例如,如果你想获取标签为'label'的列,可以使用以下代码:
```python
import csv
filename = "./dataset/dataTime2.csv"
list1 = \[\]
with open(filename, 'r') as file:
reader = csv.DictReader(file)
column = \[row\['label'\] for row in reader\]
```
这样,column列表中就包含了标签为'label'的列中的所有数值。
方法二:使用csv.reader
另一种方法是使用csv.reader来读取CSV文件,并通过索引来获取指定列的数值。你可以使用以下代码示例来提取CSV文件中的第一列和第三列:
```python
import csv
filename = "./dataset/dataTime2.csv"
list1 = \[\]
with open(filename, 'r') as f:
reader = csv.reader(f)
for row in reader:
col1 = row\[0\] # 第一列数据
col3 = row\[2\] # 第三列数据
# 处理数据 ...
```
在这个示例中,col1变量将包含每一行的第一列数据,col3变量将包含每一行的第三列数据。你可以根据需要进行进一步的处理。
方法三:处理成list\[np.array\]形式的数据
如果你想将CSV文件的数据处理成list\[np.array\]的形式,你可以使用以下代码示例:
```python
import csv
import numpy as np
filename = "./dataset/dataTime2.csv"
list1 = \[\]
with open(filename, 'r') as file:
a = file.readline()
while a:
c = np.array(a.strip("\n").split(","))
list1.append(c)
```
在这个示例中,每一行的数据将被处理成一个numpy数组,并添加到list1列表中。你可以根据需要进一步处理这个列表。
希望这些方法能够帮助到你!
#### 引用[.reference_title]
- *1* *3* [用python获取.csv文件中某一列或者某些列](https://blog.csdn.net/weixin_64338372/article/details/130251739)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v91^control_2,239^v3^insert_chatgpt"}} ] [.reference_item]
- *2* [python 提取csv文件某几列](https://blog.csdn.net/weixin_35753291/article/details/129069565)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v91^control_2,239^v3^insert_chatgpt"}} ] [.reference_item]
[ .reference_list ]
阅读全文