Python脚本筛选两列数据中相关的值

Question

怎么用Python脚本筛选两列数据中相关的值
比如像这样
1 2
1 3
4 1
2 5
6 3
7 8
最后得到1 2 3 4 5 6 一组数，7 8一组数
运用字典筛选，编不出来，希望能得到大家的指点思路

Answer 1

可以使用Python中的字典来实现对两列数据的筛选。具体步骤如下：

1. 读取数据，将其保存为列表。

   data = []
   with open("data.txt", "r") as f:
    for line in f:
        data.append(line.strip().split())
   

2. 构建一个字典，将第一列作为键，第二列作为值，并将值保存为列表。

   dict_data = {}
   for d in data:
    if d[0] not in dict_data:
        dict_data[d[0]] = [d[1]]
    else:
        dict_data[d[0]].append(d[1])
   

3. 遍历字典，将相关的值保存到一个列表中，并输出结果。

   result = []
   for key, value in dict_data.items():
    if len(value) > 1:
        result.extend([key] + value)
    else:
        result.append(key + " " + value[0])
   print(result)
   

完整代码如下：

data = []
with open("data.txt", "r") as f:
    for line in f:
        data.append(line.strip().split())

dict_data = {}
for d in data:
    if d[0] not in dict_data:
        dict_data[d[0]] = [d[1]]
    else:
        dict_data[d[0]].append(d[1])

result = []
for key, value in dict_data.items():
    if len(value) > 1:
        result.extend([key] + value)
    else:
        result.append(key + " " + value[0])
print(result)

假设数据保存在data.txt文件中，文件内容如下：

1 2
1 3
4 1
2 5
6 3
7 8

运行以上代码，将输出以下结果：

['1', '2', '3', '4', '5', '6', '7 8']

其中，结果中的每一组数表示相关的数据，例如"1 2 3"表示第一列为1的数据中，第二列为2和3的数据。最后一组"7 8"表示第一列为7的数据中，第二列为8的数据。