Question

我有一个csv文件，其结构如下：

A, 10
B, 11
C, 8
D, 12
A, 21
B, 7
D, 22
D, 15
C, 111
D, 50
A, 41
B, 32
C, 19
D, 11

我想读取整个文件，如果行是A，B，C，D格式，则将数据保存在第二列。我有一个清单：

my_list = [A, B, C, D]

我检查每4行，如果是my_list格式，则读取并保存到数据库中，如：

with open('csv_file.csv', 'rb') as csvfile:
        the_file = csv.reader(csvfile.read().splitlines())

        for idx, row in enumerate(islice(zip(*[the_file]*4), 100)):
            my_model = Django_model()
            if row[0][0] == my_list[0]:
                if row[0][0] == my_list[0] and row[1][0] == my_list[1] and \
                    row[2][0] == my_list[2] and row[3][0] == my_list[3]:
                    my_model.a_field = row[0][1]
                    my_model.b_field = row[1][1]
                    my_model.c_field = row[2][1]
                    my_model.d_field = row[3][1]
                    my_model.save()

事实是，只有当row的结构与my_list相同时，它才有效。但当它到达凌乱的部分（A，B，D，D，C，D）时，它将无法正确读取行，因此会跳过许多行。

问题是，如何跳转到下一个有趣的行（符合my_list格式）并阅读它？同时将跳过的行保存在另一个列表中？

我听说熊猫可以提供帮助，但我已经通过了文档，我找不到解决此案的方法。

Answer 1

您可以像这样提取模式和相应的值：

import pandas as pd
import re

df = pd.read_csv('/home/yusuf/Desktop/c1', header=None, names=['c1','c2'])
l1=[]
for a in re.finditer('ABCD', ''.join(df.c1.tolist())):
    l1.append(range(a.start(),a.end()))
l2 = [b for a in l1 for b in a]
print df[df.index.isin(l2)], '\n'
print df[~df.index.isin(l2)]

输出：

Python / Django-用乱码行保存csv文件数据

1 个答案: