使用 python 在 CSV 文件中的现有行之间插入一行

Question

我正在创建一个脚本，该脚本使用 Python 将行写入 CSV 文件。

现在，我的脚本以这种格式编写 CSV：

Title row 
Value1;Value2;.... (more than 70)
Title row2 
Value1;Value2;...

我只想能够再次读取文件并在行之间插入一行值，如下所示：

Title row 
Value1;Value2;.... (more than 70)
Value1;Value2;....
Title row2 
Value1;Value2;...

你有什么想法？

Answer 1

我认为您可以尝试使用 Header Titles 获取行的索引，然后使用 append 添加新行，然后再次组合两个数据帧。 这是可能对您有用的代码。

import pandas as pd 

# intialise data of lists. 
data = {'Title':['Tom', 'nick', 'krish', 'jack','Title','Harry'], 
        'Row':[20, 21, 19, 18,'Row',21]}
new_data = {'Title':['Rahib'], 
        'Row':[25]}     
  
# Create DataFrame 
df = pd.DataFrame(data) 
new_df = pd.DataFrame(new_data)
#print(df)

index = df[df['Title'] == 'Title'].index.values.astype(int)[0]

upper_df = df.loc[:index-1]
lower_df = df.loc[index+1:]

upper_df = upper_df.append(new_df)
upper_df = upper_df.append(lower_df).reset_index(drop=True)

print(upper_df)

这将在 dataframe 之后返回。

   Title Row
0    Tom  20
1   nick  21
2  krish  19
3   jack  18
4  Rahib  25
5  Harry  21

Answer 2

import csv

with open('csvfile.csv', mode='w') as csv_file:
    fieldnames = ['Title', 'row']
    writer = csv.DictWriter(csv_file, delimiter=';',fieldnames=fieldnames)
    writer.writeheader()
    writer.writerow({'Title': 'Value1', 'row': 'Value2'})

Answer 3

感谢@harshal 的回答和@Rahib 的回答，但我已经尝试 pandas 很长时间了，似乎无法让它工作，看起来它不太适合我的 CSV 格式。

最后，我查看了@Serge Ballesta 提供的帖子，实际上，一个简单的readlines和行索引的检索是一个非常简单的技巧：

with open(output) as myFile:
    for num, line in enumerate(myFile, 1):
        if lookup in line:
            index = num

f = open(output, "r")
contents = f.readlines()

value=';'.join(value)
f.close()
contents.insert(index+1, str(value)+'\n')

f = open(output, "w")
contents = "".join(contents)
f.write(contents)
f.close()

output是文件名（传入参数）， value是值列表（以“;”作为分隔符加入字符串）， lookup是i正在寻找的字符串（标题行）