如何将字符串添加到csv中的列

Question

我在in.csv 中有一个包含 5 列的 csv，我想在同一个文件中将第 2 列中的所有数据添加到“文本”中，以便该行

data1 data2 data3 data4 data5

变成

data1 textdata2 data3 data4 data5

我认为使用正则表达式可能是个好主意，但我不确定如何继续

编辑：

按照bigbounty的回答进行后，我使用了以下脚本：

import pandas as pd
df = pd.read_csv("in.csv")
df["id_str"] = str("text" + str(df["id_str"]))
df.to_csv("new_in.csv", index=False)

我的in.out 文件是这样的：

s_no,id_str,screen_name
1,1.15017060743203E+018,lorem
2,1.15015544419693E+018,ipsum
3,1.15015089995785E+018,dolor
4,1.15015054311063E+018,sit

运行脚本后new_in.csv文件为：

s_no,id_str,screen_name
1,"text0    1.150171e+18
1    1.150155e+18
2    1.150151e+18
3    1.150151e+18
Name: id_str, dtype: float64",lorem
2,"text0    1.150171e+18
1    1.150155e+18
2    1.150151e+18
3    1.150151e+18
Name: id_str, dtype: float64",ipsum
3,"text0    1.150171e+18
1    1.150155e+18
2    1.150151e+18
3    1.150151e+18
Name: id_str, dtype: float64",dolor
4,"text0    1.150171e+18
1    1.150155e+18
2    1.150151e+18
3    1.150151e+18
Name: id_str, dtype: float64",sit

应该是这样的：

s_no,id_str,screen_name
1,text1.15017060743203E+018,lorem
2,text1.15015544419693E+018,ipsum
3,text1.15015089995785E+018,dolor
4,text1.15015054311063E+018,sit

Answer 1

• 使用csv 模块

import csv

with open('test.csv', 'r+', newline='') as f:
    data = list(csv.reader(f))  # produces a list of lists
    for i, r in enumerate(data):
        if i > 0:  # presumes the first list is a header and skips it
            r[1] = 'text' + r[1]  # add text to the front of the text at index 1
    f.seek(0)  # find the beginning of the file
    writer = csv.writer(f)  
    writer.writerows(data)  # write the new data back to the file

# the resulting text file
s_no,id_str,screen_name
1,text1.15017060743203E+018,lorem
2,text1.15015544419693E+018,ipsum
3,text1.15015089995785E+018,dolor
4,text1.15015054311063E+018,sit

• 使用熊猫
• 此解决方案与任何列名无关，因为它使用列索引。
  • pandas.DataFrame.iloc

import pandas as pd

# read the file set the column at index 1 as str
df = pd.read_csv('test.csv', dtype={1: str})

# add text to the column at index 1
df.iloc[:, 1] = 'text' + df.iloc[:, 1]

# write to csv
df.to_csv('test.csv', index=False)

# resulting csv
s_no,id_str,screen_name
1,text1.15017060743203E+018,lorem
2,text1.15015544419693E+018,ipsum
3,text1.15015089995785E+018,dolor
4,text1.15015054311063E+018,sit

Answer 2

这可以使用 pandas 轻松完成。

import pandas as pd
df = pd.read_csv("in.csv")
df["data2"] = "text" + df["data2"].astype(str)
df.to_csv("new_in.csv", index=False)