python新手问题：将代码转换为类

Question

我有这个代码：

import csv
import collections

def do_work():
      (data,counter)=get_file('thefile.csv')
      b=samples_subset1(data, counter,'/pythonwork/samples_subset3.csv',500)
      return

def get_file(start_file):

        with open(start_file, 'rb') as f:
            data = list(csv.reader(f))
            counter = collections.defaultdict(int)

            for row in data:
              counter[row[10]] += 1
            return (data,counter)

def samples_subset1(data,counter,output_file,sample_cutoff):

      with open(output_file, 'wb') as outfile:
          writer = csv.writer(outfile)
          b_counter=0
          b=[]
          for row in data:
              if counter[row[10]] >= sample_cutoff:
                 b.append(row) 
                 writer.writerow(row)
                 b_counter+=1
      return (b)

我最近开始学习Python，并且想从良好的习惯开始。因此，我想知道您是否可以帮助我开始将此代码转换为类。我不知道从哪里开始。

Answer 1

根据我对原始帖子的评论，我认为这里不需要上课。不过，如果其他 Python 程序员会读到这篇文章，我建议将其与 PEP8（Python 风格指南）内联。这是一个快速重写：

import csv
import collections

def do_work():
    data, counter = get_file('thefile.csv')
    b = samples_subset1(data, counter, '/pythonwork/samples_subset3.csv', 500)

def get_file(start_file):
    with open(start_file, 'rb') as f:
        counter = collections.defaultdict(int)
        data = list(csv.reader(f))

        for row in data:
            counter[row[10]] += 1

    return (data, counter)

def samples_subset1(data, counter, output_file, sample_cutoff):
    with open(output_file, 'wb') as outfile:
        writer = csv.writer(outfile)
        b = []
        for row in data:
            if counter[row[10]] >= sample_cutoff:
                b.append(row) 
                writer.writerow(row)

    return b

注意：

1. 没有人使用超过 4 个空格
   永远缩进。使用 2 - 4。以及所有
   你的缩进级别应该
   匹配。
2. 参数之间的逗号后使用一个空格
   到函数（“F（a，b，c）”不
   "F(a,b,c)")
3. 函数末尾的裸返回语句
   是没有意义的。功能无
   return 语句隐式返回
   无
4. 全部周围有一个空格
   （a = 1，而不是 a=1）
5. 运算符
   将单个值括在括号中。
   它看起来像一个元组，但事实并非如此。
6. b_counter 根本没有使用，所以我
   将其删除。
7. csv.reader 返回一个迭代器，您将其转换为列表。这通常是一个坏主意，因为它迫使 Python 立即将整个文件加载到内存中，而迭代器只会根据需要返回每一行。理解迭代器对于编写高效的 Python 代码绝对必要。我暂时保留了 data，但您可以重写以在使用 data（它是一个列表）的任何地方使用迭代器。

Answer 2

好吧，我不确定你想把什么变成一个类。你知道什么是班级吗？您想要创建一个类来表示某种类型的事物。如果我正确理解您的代码，您希望过滤 CSV 以仅显示那些 row[ 10 ] 至少被 sample_cutoff 其他行共享的行。当然，使用 Excel 过滤器可以比在 Python 中读取文件更容易做到这一点吗？

其他线程中的人建议的内容是正确的，但并不真正适用于您的情况。您不必要地使用了很多全局变量：如果它们对于代码来说是必需的，您应该将所有内容放入类中并将它们设为属性，但是由于您一开始就不需要它们，因此没有必要创建班级。

关于代码的一些提示：

• 不要将文件转换为列表。这使得 Python 将整个内容一次性读入内存，如果你有一个大文件，这会很糟糕。相反，只需迭代文件本身：for row in csv.reader(f): 然后，当您想再次浏览文件时，只需执行 < code>f.seek(0) 返回顶部并重新开始。

• 不要将return放在每个函数的末尾；那是没有必要的。您也不需要括号：返回垃圾邮件 就可以。

改写

import csv
import collections

def do_work():
    with open( 'thefile.csv' ) as f:
        # Open the file and count the rows.
        data, counter = get_file(f)
        
        # Go back to the start of the file.
        f.seek(0)

        # Filter to only common rows.
        b = samples_subset1(data, counter, 
            '/pythonwork/samples_subset3.csv', 500)
   
     return b

def get_file(f):
    counter = collections.defaultdict(int)
    data = csv.reader(f)
    
    for row in data:
        counter[row[10]] += 1

    return data, counter

def samples_subset1(data, counter, output_file, sample_cutoff):
    with open(output_file, 'wb') as outfile:
        writer = csv.writer(outfile)
        b = []
        for row in data:
            if counter[row[10]] >= sample_cutoff:
                b.append(row) 
                writer.writerow(row)

    return b