Question

我需要定义一个读取csv文件的函数，然后我需要做一些从所选文件中获取信息并进行一些计算的函数，然后将其写入新文件。这是代码的一部分：

if str1 == 'small':
    with open('small.csv', 'r+') as small:
        read_small = csv.reader(small)
        for row in read_small:
            print row
elif str1 == 'big':
    with open('big.csv', 'r+') as big:
            read_big = csv.reader(big)
            for row in read_big:
                print row    
else:
    print "File not found"

现在，它打印出我希望它们看的文件，但是当我尝试def一个不打印文件的函数时，只需为它返回它，这样我就可以在其他函数中调用它来进行计算不行！有小费吗？（我用google搜索****以找到解决方案，我是新手）

Answer 1

但是当我尝试def一个不打印文件的函数时，只需为我返回

def get_lines(my_str):
   file_picker = {'small': 'small.csv', 'big':'big.csv'}
   with open(file_picker[my_str]) as f:
      my_csv_reader = csv.reader(f)
      return [ row for row in my_csv_reader ]

Answer 2

听起来你想要找回所有行的列表。

def get_rows():
    rows_to_return = []
    with open('file.csv', 'r') as f:
        reader = csv.reader(f)
        for row in reader:
            rows_to_return.append(row)
    return rows_to_return

从文件读取和写入新文件的功能

2 个答案: