1

我需要一些关于将文本/字符串写入 csv 文件的建议。

以下是我的程序代码:

import pandas as pd

df = pd.read_csv('id.csv', index_col=('NAME'))
flag = True

print(df)
    
while(flag):
    id=str(input("Enter identification: ")[:9]).upper()
    particulars = str(df[df["identification"] == id])
    if (df[df["identification"] == id].empty):
        print("No records found, try again")
        continue
    else:
        print(particulars)
        f = open("today.csv", "w", newline="")
        f.write(particulars)
        continue

对于 IF 语句,当键入的标识不匹配或在 id.csv 文件中找不到时,我能够实现“未找到记录”

如果标识可用,它将打印出相关信息并将信息写入today.csv文件。

接下来我想做的是:

  1. 如果在 id.csv 中找到标识,则将提取的信息写入

  2. 写入提取的信息后,循环将重新开始,当键入另一个标识时,它将继续将相关信息写入同一个 csv 文件(today.csv)

现在,我可以读(id.csv)和写(today.csv)。但我只能将第一个标识(及其相关信息)写入 csv 文件。

我很确定问题出在我的写入功能上。

谁能指出我正确的方向来让这件事运行起来?

4

0 回答 0