在查找包含数组中值的 CSV 行时需要帮助

Question

我有一个数组 LiveTick = ['ted3m index','US0003m index','USGG3m index']，我正在读取一个 CSV 文件 book1.csv。我必须找到包含 csv 中的值的行。

例如，第 15 行将包含 ted3m index 500 | 600，第 20 行将包含 US0003m index 800 | 900，以此类推。

然后我必须获取行中包含的值，并为数组LiveTick 中包含的每个值解析它。我该如何进行？下面是我的示例代码：

 with open('C:\\blp\\book1.csv', 'r') as f:
   reader = csv.reader(f, delimiter=',')
   writer = csv.writer(outf)
   for row in reader:
      for list in LiveTick:
         if list in row:
            print ('Found: {}'.format(row))

Answer 1

您可以使用 pandas，它的速度非常快，并且开箱即用地为您完成所有读取、写入和过滤工作：

import pandas as pd

df = pd.read_csv('C:\\blp\\book1.csv')
filtered_df = df[df['your_column_name'].isin(LiveTick)]

# now you can save it
filtered_df.to_csv('C:\\blp\\book_filtered.csv')

Answer 2

您的想法是正确的，但您可以进行一些改进：

• 不要使用不会短路的嵌套for 循环，而是使用any 将第一列与多个值进行比较。
• 在进行过程中写入您的 csv，而不仅仅是 print。这样可以节省内存，因为您每次只能在内存中保存一行。
• 在with 语句中将outf 定义为open 对象。
• 不要隐藏内置list。使用另一个标识符，例如i，用于LiveTick 中的元素。

这是一个演示：

with open('in.csv', 'r') as f, open('out.csv', 'wb', newline='') as outf:
    reader = csv.reader(f, delimiter=',')
    writer = csv.writer(outf, delimiter=',')
    for row in reader:
        if any(i in row[0] for i in LiveTick):
            writer.writerow(row)