我有以下代码:
filename=input("Name of file?")
sheet=input("Name of sheet?")
df=pd.read_excel('C:/Users/Path/JAN.xls',
sheetname=sheet,
header=0,
parse_cols="C:BD",
na_values='T')
我希望文件名的"JAN"部分在文件运行之前可以根据用户输入进行更改。例如,我在同一目录中有其他文件标记为"/JAN"、"/FEB"、"/MAR"等。如何添加可以根据用户输入更改文件 io 的变量?
input= "C:/Users/Path/" + user_input + ".xls"
df = pd.read_excel(filename,
sheetname=sheet,
header=0,
parse_cols="C:BD",
na_values='T')
或者我所做的是,我使用了图书馆 glob
files = glob.glob('C:/Users/Path/*')
for file in files:
input = input('Use ' + file + '?')
if input == y or input == yes:
sheet = input('Sheet name?')
df = pd.read_excel(filename,
sheetname=sheet,
header=0,
parse_cols="C:BD",
na_values='T')