import os
import csv
lst =[]
# 将所有文件的路径放入到listcsv列表中
def list_dir(file_dir):
# list_csv = []
dir_list = os.listdir(file_dir)
for cur_file in dir_list:
path = os.path.join(file_dir, cur_file)
# 判断是文件夹还是文件
if os.path.isfile(path):
# print("{0} : is file!".format(cur_file))
dir_files = os.path.join(file_dir, cur_file)
# 判断是否存在.csv文件,如果存在则获取路径信息写入到list_csv列表中
if os.path.splitext(path)[1] == '.csv':
csv_file = os.path.join(file_dir, cur_file)
# print(os.path.join(file_dir, cur_file))
# print(csv_file)
list_csv.append(csv_file)
if os.path.isdir(path):
# print("{0} : is dir".format(cur_file))
# print(os.path.join(file_dir, cur_file))
list_dir(path)
return list_csv
if __name__ == '__main__':
paths = r'C:\Users\Administrator\Desktop\AAA'
list_csv = []
list_dir(file_dir=paths)
print(list_csv)
import csv
lst = []
with open('data.csv', 'r', encoding='utf-8') as csvfile:
reader = csv.reader(csvfile)
for row in reader:
print(row) # 输出data文件内容
n = len(row)
lst.append(row[1])
print(lst[-1])
用pandas不香吗?几行代码就搞定了。。
我感觉你应该遍历文件夹里面的csv文件 循环读取然后再进行逻辑处理
既然读取代码有了,定义一个读取函数,遍历文件直接读取应该可以
import os
import csv
def print_out(file_name):
lst = []
with open(file_name, 'r', encoding='utf-8') as csvfile:
reader = csv.reader(csvfile)
for row in reader:
print(row) # 输出data文件内容
n = len(row)
lst.append(row[1])
print(lst[-1])
# 将所有文件的路径放入到listcsv列表中
def list_dir(file_dir):
# list_csv = []
dir_list = os.listdir(file_dir)
for cur_file in dir_list:
path = os.path.join(file_dir, cur_file)
# 判断是文件夹还是文件
if os.path.isfile(path):
# print("{0} : is file!".format(cur_file))
dir_files = os.path.join(file_dir, cur_file)
# 判断是否存在.csv文件,如果存在则获取路径信息写入到list_csv列表中
if os.path.splitext(path)[1] == '.csv':
csv_file = os.path.join(file_dir, cur_file)
# print(os.path.join(file_dir, cur_file))
# print(csv_file)
list_csv.append(csv_file)
if os.path.isdir(path):
# print("{0} : is dir".format(cur_file))
# print(os.path.join(file_dir, cur_file))
list_dir(path)
return list_csv
if __name__ == '__main__':
paths = r'C:\Users\Administrator\Desktop\AAA'
list_csv = []
list_dir(file_dir=paths)
print(list_csv)
for path in list_csv:
print_out(path)