0

这是我的代码，它计算唯一数字出现的次数，并在列表 A 中查找重复 5 次的数字，然后在列表 A 中重复 2 次的数字，并打印列表 A 的任何匹配结果与相应的值来自列表 B。此外，列表 A 和 B 的大小始终相同。

a = (['12','12','12','12','12','23','24','24','31','31'])
b = (['1','2','2','2','2','2','5','5','5','5'])

from collections import Counter
counts = Counter(a)

c = []

for ai , bi in zip(a,b):
   if counts[ai] == 5:
       c.append([ai,bi])
   elif counts[ai] == 1:
       c.append([ai,bi])
   else:
       None

print(c)
#[['12', '1'], ['12', '2'], ['12', '2'], ['12', '2'], ['12', '2'], ['23', '2']]

有没有一种快速的方法可以让我的代码将多个输出列表重新格式化为如下所示的列表：

#[('12', '12', '12', '12', '12', '23'), ('1', '2', '2', '2', '2', '2')]

这样每个列表中的每一列都可以有自己的列表。

谢谢！

4

3 回答 3

2

那个怎么样：

import itertools

c = [['12', '1'], ['12', '2'], ['12', '2'], ['12', '2'], ['12', '2'], ['23', '2']]

merged = list(itertools.chain(*c))
# merged = ['12', '1', '12', '2', '12', '2', '12', '2', '12', '2', '23', '2']

split = [tuple(merged[::2]), tuple(merged[1::2])]
# split = [('12', '12', '12', '12', '12', '23'), ('1', '2', '2', '2', '2', '2')]
于 2017-08-22T13:49:50.033 回答
1

从您的代码继续

c = [['12', '1'], ['12', '2'], ['12', '2'], ['12', '2'], ['12', '2'], ['23', '2']]

cols = [] #initialize list of columns
for i in range(2):
  tup = tuple([item[i] for item in c]) #create tuple from loop of column
  cols.append(tup)
print(cols) # [('12', '12', '12', '12', '12', '23'), ('1', '2', '2', '2', '2', '2')]
于 2017-08-22T14:11:04.463 回答
1

您可以使用列表理解

>>> x = [['12', '1'], ['12', '2'], ['12', '2'], ['12', '2'], ['12', '2'], ['23', '2']]
>>> [tuple(v[0] for v in x), tuple(v[1] for v in x)]
[('12', '12', '12', '12', '12', '23'), ('1', '2', '2', '2', '2', '2')]

以您最初想要的格式保存数据可能会更有效。元组是不可变的，因此使用列表并附加到它们会更容易。就像是：

columns = []
values = []

for ai , bi in zip(a,b):
   if counts[ai] == 5:
       columns.append(ai)
       values.append(bi)
   elif counts[ai] == 1:
       columns.append(ai)
       values.append(bi)

print([columns, values])

# prints this
[['12', '12', '12', '12', '12', '23'], ['1', '2', '2', '2', '2', '2']]
于 2017-08-22T14:14:16.580 回答