怎么读取csv文件,提取一列内容,并写入新的一列
时间: 2023-05-26 09:01:54 浏览: 46
您可以使用Python中的pandas库来读取和操作CSV文件。
首先,您需要使用pandas.read_csv()函数读取CSV文件,例如:
```python
import pandas as pd
df = pd.read_csv('yourfile.csv')
```
接下来,您可以使用df[‘column_name’]来提取特定列的内容。例如:
```python
col = df['your_column_name']
```
最后,您可以使用df[‘new_column_name’] = col来将提取的列内容写入新的一列。例如:
```python
df[‘new_column_name’] = col
```
完整代码示例如下:
```python
import pandas as pd
# 读取CSV文件
df = pd.read_csv('yourfile.csv')
# 提取特定列内容
col = df['your_column_name']
# 将提取的内容写入新的一列
df[‘new_column_name’] = col
```
请替换'yourfile.csv','your_column_name'和'new_column_name'为您自己的文件和列名。
相关问题
pandas读取csv指定列的内容写入txt文件固定位置。
可以按照以下步骤实现:
1. 使用pandas读取csv文件,指定需要提取的列。
```python
import pandas as pd
# 读取csv文件,指定需要提取的列
df = pd.read_csv('data.csv', usecols=['column1', 'column2'])
```
2. 将提取的列内容写入txt文件。
```python
# 将提取的列内容写入txt文件
with open('output.txt', 'w') as f:
for row in df.iterrows():
f.write('{} {}\n'.format(row[1]['column1'], row[1]['column2']))
```
其中,`with open('output.txt', 'w') as f:` 表示打开一个名为`output.txt`的文件进行写入操作,`'w'`表示以写入方式打开文件。`for row in df.iterrows():` 表示遍历`df`中的每一行数据,`row`为元组类型,`row[1]`表示该行数据,`['column1']`和`['column2']`表示需要提取的列。`f.write('{} {}\n'.format(row[1]['column1'], row[1]['column2']))` 表示将提取的列内容按照指定格式写入txt文件,`'\n'`表示换行符。
python提取csv文件的某一列内容保存到txt文件
可以使用Python的csv模块读取csv文件,然后将指定列的内容保存到txt文件中。
以下是一个示例代码:
```python
import csv
# 读取csv文件
with open('data.csv', 'r') as csvfile:
reader = csv.reader(csvfile)
# 获取指定列的内容
column = [row[2] for row in reader]
# 将内容保存到txt文件
with open('column.txt', 'w') as txtfile:
txtfile.write('\n'.join(column))
```
上述代码中,我们打开了一个名为"data.csv"的csv文件,并使用csv.reader对象读取其中的内容。然后,我们使用列表推导式获取了第三列的内容,并将其存储在名为"column"的列表中。
接下来,我们打开一个名为"column.txt"的txt文件,并使用文件对象的write方法将"column"列表中的内容写入文件中。我们使用join方法将列表中的元素连接成一个字符串,并在每个元素之间添加换行符,以使每个元素占用一行。