如何创建用于选择要访问的不同文件的输入？

如何解决如何创建用于选择要访问的不同文件的输入？

我对python很陌生，所以请多多包涵。

当前，这是我的代码：

import pandas as pd
import statistics
import matplotlib.pyplot as plt
import math
from datetime import datetime

start_time = datetime.now()


gf = pd.read_csv(r"/Users/aaronhuang/Documents/Desktop/ffp/exfileCLEAN2.csv",skiprows=[1])
bf = pd.read_csv(r"/Users/aaronhuang/Documents/Desktop/ffp/2SeconddatasetCLEAN.csv",skiprows=[1])

df = (input("Which data set? "))

magnitudes = (df['Magnitude '].values)
times = df['Time '].values
average = statistics.mean(magnitudes)
sd = statistics.stdev(magnitudes)
below = sd * 3


class data_set:
    def __init__(self,index):
        self.mags = []
        self.i = index
        self.mid_time = df['Time '][index]
        self.mid_mag = df['Magnitude '][index]
        self.times = []
        ran = 80
        for ii in range(ran):
            self.times.append(df['Time '][self.i + ii - ran / 2])
            self.mags.append(df['Magnitude '][self.i + ii - ran / 2])


data = []


today = float(input("What is the range? "))

i = 0
while (i < len(df['Magnitude '])):
    if (abs(df['Magnitude '][i]) <= (average - below)):
        # check if neighbours
        t = df['Time '][i]

        tt = True
        for d in range(len(data)):
            if abs(t - data[d].mid_time) <= today:
                # check if closer to center
                if df['Magnitude '][i] < data[d].mid_mag:
                    data[d] = data_set(i)
                    print("here")
                tt = False
                break
        if tt:
            data.append(data_set(i))
    i += 1
print("found values")

# graphing


height = 2  # Change this for number of columns
width = math.ceil(len(data) / height)
if width < 2:
    width = 2
fig,axes = plt.subplots(width,height,figsize=(30,30))

row = 0
col = 0

for i in range(len(data)):

    axes[row][col].plot(data[i].times,data[i].mags)
    col += 1
    if col > height - 1:
        col = 0
        row += 1
plt.show()


end_time = datetime.now()
print('Duration: {}'.format(end_time - start_time))

当前，产生的错误是这样的：

/Users/aaronhuang/.conda/envs/EXTTEst/bin/python "/Users/aaronhuang/PycharmProjects/EXTTEst/Code sandbox.py"
Which data set? gf
Traceback (most recent call last):
  File "/Users/aaronhuang/PycharmProjects/EXTTEst/Code sandbox.py",line 14,in <module>
    magnitudes = int(df['Magnitude '].values)
TypeError: string indices must be integers

Process finished with exit code 1

我试图让用户能够选择要访问的文件以执行其余代码。

因此，如果用户键入gf，我希望代码访问第一个数据文件。

任何帮助将不胜感激。谢谢

解决方法

为什么不一开始就使用if语句？试试这个：

代替：

gf = pd.read_csv(r"/Users/aaronhuang/Documents/Desktop/ffp/exfileCLEAN2.csv",skiprows=[1])
bf = pd.read_csv(r"/Users/aaronhuang/Documents/Desktop/ffp/2SeconddatasetCLEAN.csv",skiprows=[1])

df = (input("Which data set? "))

使用此：

choice = input("Which data set? ")

if choice == "gf":
    df = pd.read_csv(r"/Users/aaronhuang/Documents/Desktop/ffp/exfileCLEAN2.csv",skiprows=[1])
elif choice == "bf":
    df = pd.read_csv(r"/Users/aaronhuang/Documents/Desktop/ffp/2SeconddatasetCLEAN.csv",skiprows=[1])
else:
    print("Error. Your choice is not valid")
    df = ""
    break