如何在一段时间循环之前一劳永逸地提示用户输入？

Question

我正在编写一个Python脚本，它从名为“read_file”的熊猫数据帧中获取特定数据，并将其写入csv文件。为此，我循环遍历DataFrame的行，直到满足某个条件，也就是说，我将数据放入字典中，然后使用“To_csv”函数更新csv文件。当然，执行同一件事情的方法更少，也更简单，但我将把优化留到另一天（尽管我愿意接受建议）。

我想知道是否有一种方法可以提示用户决定他们想要写数据的方式。基本上，我是在问他们是否愿意:

1. 覆盖文件，然后通过to_csv函数的追加模式逐行更新它（每次我们通过循环）
2. 不覆盖文件，只是在每次运行脚本时向其附加数据（这意味着它在运行后变得更大）。

到目前为止，我的代码如下所示：

i = 0
for index, row in read_file.iterrows():
case = row['Case']
first = case.split('-')[0]
second = case.split('-')[1]
third = case.split('-')[2]
fourth = case.split('-')[3]
fifth = case.split('-')[4]
if first == 'X01': # if1
    if second == '01': # if2
        if fourth == '04': # if3
            i += 1
            Ax = float(row['Ax'])
            Ay = float(row['Ay'])
            Az = float(row['Az'])
            ENT = float(row['ENT'])
            Ips = (Ax**2 + Ay**2 + Az**2)**(0.5)
            beta = float(row['beta'])
            date = row['Date'].replace("/", "-")
            totalP = float(row['Total.P'])

            data = pd.DataFrame({'Case': [str(case)],
                'ENT': [ENT],
                'total P': [totalP]},
                index = [i])

            filename = 'curve_{}-{}-{}-{}_I{}-B{}-D{}.csv'.format(first,second,third,fourth,round(Ips, 2),beta,date)
            if os.path.getsize(filename):
                print('Curve file not empty.')
                while True:
                    inp = input('Do you want to: A) Append the file. B) Overwrite the file. [A/B]? : ')
                    if inp in ['A', 'B']:
                        break
                if inp == 'A':
                    print('Appending... ')
                    data.to_csv(filename, mode='a')
                elif inp == 'B':
                    print('Overwriting... ')
                    data.to_csv(filename, mode='w')

但是，每次条件 if1、if2

谢谢

编辑：

示例文件为：

预期输出为：

Answer 1

在循环访问文件之前向用户询问问题，这样问题只会被问（和回答）一次：

while True:
    inp = input('Do you want to: A) Append the file. B) Overwrite the file. [A/B]? : ')
    if inp in ['A', 'B']:
        break

i = 0
for index, row in read_file.iterrows():
case = row['Case']
first = case.split('-')[0]
second = case.split('-')[1]
third = case.split('-')[2]
fourth = case.split('-')[3]
fifth = case.split('-')[4]
if first == 'X01': # if1
    if second == '01': # if2
        if fourth == '04': # if3
            i += 1
            Ax = float(row['Ax'])
            Ay = float(row['Ay'])
            Az = float(row['Az'])
            ENT = float(row['ENT'])
            Ips = (Ax**2 + Ay**2 + Az**2)**(0.5)
            beta = float(row['beta'])
            date = row['Date'].replace("/", "-")
            totalP = float(row['Total.P'])

            data = pd.DataFrame({'Case': [str(case)],
                'ENT': [ENT],
                'total P': [totalP]},
                index = [i])

            filename = 'curve_{}-{}-{}-{}_I{}-B{}-D{}.csv'.format(first,second,third,fourth,round(Ips, 2),beta,date)
            if os.path.getsize(filename):
                print('Curve file not empty.')
                if inp == 'A':
                    print('Appending... ')
                    data.to_csv(filename, mode='a')
                elif inp == 'B':
                    print('Overwriting... ')
                    data.to_csv(filename, mode='w')