我有一个程序在1个excel csv文件中搜索A列中的日期并返回该搜索的行。我现在想要扩展和搜索日期数组并搜索该日期的各种excel文件。
所以,例如。我的输入数组是:
[11202004,11212006,2192008,6102001]
然后程序应搜索那些日期的所有excel csv文件(date1.csv,date2.csv,date3.csv等)并返回包含行数据的数组数组。
输出应为:
[[11202004,111,222,333],[11212006,444,555,666],[2192008,777,888,999],[6102001,120,234,566]]
我的代码:
date_to_find = '2102006'
x = 1
myarr = range(20)
while(x < 20):
with open('date'+str(x)+'.csv', 'r') as f:
myarr[x] = next((line for line in csv.reader(f) if line[0] == date_to_find),'')
x += 1
return myarr