Python:append()和extend()

时间:2019-07-11 03:30:17

标签: python regex pandas

我有一个300万行的.txt文件。该文件包含的数据如下所示:

# RSYNC: 0 1 1 0 512 0
#$SOA 5m localhost. hostmaster.localhost. 1906022338 1h 10m 5d 1s
# random_number_ofspaces_before_this text $TTL 60s
#more random information
:127.0.1.2:https://www.spamhaus.org/query/domain/$
test
:127.0.1.2:https://www.spamhaus.org/query/domain/$
.0-0m5tk.com
.0-1-hub.com
.zzzy1129.cn
:127.0.1.4:https://www.spamhaus.org/query/domain/$
.0-il.ml
.005verf-desj.com
.01accesfunds.com

我正在尝试解析它,使其看起来像:

+--------------------+--------------+-------------+-----------------------------------------------------+
|    domain_name     | period_count | parsed_code |                      raw_code                       |
+--------------------+--------------+-------------+-----------------------------------------------------+
| test               |            0 | 127.0.1.2   |  :127.0.1.2:https://www.spamhaus.org/query/domain/$ |
| .0-0m5tk.com       |            2 | 127.0.1.2   |  :127.0.1.2:https://www.spamhaus.org/query/domain/$ |
| .0-1-hub.com       |            2 | 127.0.1.2   |  :127.0.1.2:https://www.spamhaus.org/query/domain/$ |
| .zzzy1129.cn       |            2 | 127.0.1.2   |  :127.0.1.2:https://www.spamhaus.org/query/domain/$ |
| .0-il.ml           |            2 | 127.0.1.4   |  :127.0.1.4:https://www.spamhaus.org/query/domain/$ |
| .005verf-desj.com  |            2 | 127.0.1.4   |  :127.0.1.4:https://www.spamhaus.org/query/domain/$ |
| .01accesfunds.com  |            2 | 127.0.1.4   |  :127.0.1.4:https://www.spamhaus.org/query/domain/$ |
+--------------------+--------------+-------------+-----------------------------------------------------+

为此,我提出了以下建议:

rows = []
raw_code = None
parsed_code = None
with open('dbl-sr-2019-06-02T23_38_27Z.txt', 'r') as f: # assumes the file name is input.txt
    for line in f:
        line = line.rstrip('\n')
        if line.startswith(':127'):
            raw_code = line
            parsed_code = re.split(":", line)[1]
            continue
        if line.startswith('#'):
            continue
        rows.append((line, parsed_code))
#         rows.append((raw_code))
#         rows.extend((line, parsed_code, raw_code))
#         rows.extend((raw_code))

import pandas as pd 
df = pd.DataFrame(rows, columns=['domain_name', "parsed_code" 'raw_spamhaus_return_code'])
print(df)

上面的代码块中的注释行或者没有产生我想要的输出,或者给出了错误。我正在努力建立一个超过2列的Pandas数据框。我可以得到domain_name和另外一列。看来我无法获取正确使用.append.extend函数的代码。有人可以提供指导吗?

1 个答案:

答案 0 :(得分:5)

问题的可能根源是缺少逗号。

此:

df = pd.DataFrame(rows, columns=[
    'domain_name', 'parsed_code', 'raw_spamhaus_return_code'])

不同于:

df = pd.DataFrame(rows, columns=[
    'domain_name', "parsed_code" 'raw_spamhaus_return_code'])

因为(请注意缺少逗号):

"parsed_code" 'raw_spamhaus_return_code'

成为一个字符串。

测试代码:

import re

data = [x.strip() for x in """
    # RSYNC: 0 1 1 0 512 0
    #$SOA 5m localhost. hostmaster.localhost. 1906022338 1h 10m 5d 1s
    # random_number_ofspaces_before_this text $TTL 60s
    #more random information
    :127.0.1.2:https://www.spamhaus.org/query/domain/$
    test
    :127.0.1.2:https://www.spamhaus.org/query/domain/$
    .0-0m5tk.com
    .0-1-hub.com
    .zzzy1129.cn
    :127.0.1.4:https://www.spamhaus.org/query/domain/$
    .0-il.ml
    .005verf-desj.com
    .01accesfunds.com
""".split('\n')[1:-1]]

rows = []
raw_code = None
parsed_code = None
for line in data:
    line = line.rstrip('\n')
    if line.startswith(':127'):
        raw_code = line
        parsed_code = re.split(":", line)[1]
        continue
    if line.startswith('#'):
        continue
    rows.append((line, line.count('.'), parsed_code, raw_code))

import pandas as pd

df = pd.DataFrame(rows, columns=[
    'domain_name', 'period_count ', 'parsed_code',
    'raw_spamhaus_return_code'])
print(df)

结果:

         domain_name  period_count  parsed_code  \
0               test              0   127.0.1.2   
1       .0-0m5tk.com              2   127.0.1.2   
2       .0-1-hub.com              2   127.0.1.2   
3       .zzzy1129.cn              2   127.0.1.2   
4           .0-il.ml              2   127.0.1.4   
5  .005verf-desj.com              2   127.0.1.4   
6  .01accesfunds.com              2   127.0.1.4   

                            raw_spamhaus_return_code  
0  :127.0.1.2:https://www.spamhaus.org/query/doma...  
1  :127.0.1.2:https://www.spamhaus.org/query/doma...  
2  :127.0.1.2:https://www.spamhaus.org/query/doma...  
3  :127.0.1.2:https://www.spamhaus.org/query/doma...  
4  :127.0.1.4:https://www.spamhaus.org/query/doma...  
5  :127.0.1.4:https://www.spamhaus.org/query/doma...  
6  :127.0.1.4:https://www.spamhaus.org/query/doma...