6

我在数组中有两个数据集：

arr1 = [
  ['2011-10-10', 1, 1],
  ['2007-08-09', 5, 3],
  ...
]

arr2 = [
  ['2011-10-10', 3, 4],
  ['2007-09-05', 1, 1],
  ...
]

我想将它们组合成一个数组，如下所示：

arr3 = [
  ['2011-10-10', 1, 1, 3, 4],
  ...
]

我的意思是，只需将这些行与同一date列合并即可。

编辑

谢谢大家，只是为了澄清，我不需要那些没有出现在两个数组中的行，只需删除它们。

4

6 回答 6

6

以不同的方式组织数据（您可以轻松地将已有的数据转换为两个dicts）：

d1 = { '2011-10-10': [1, 1],
       '2007-08-09': [5, 3]
     }
d2 = { '2011-10-10': [3, 4],
       '2007-09-05': [1, 1]
     }

然后：

d3 = { k : d1[k] + d2[k] for k in d1 if k in d2 }
于 2013-07-16T17:16:06.907 回答
3

可能值得一提的是集合数据类型。因为他们的方法与问题的类型一致。集合运算符允许您使用全、内、外、左、右连接轻松灵活地连接集合。与字典一样，集合不保留顺序，但如果将集合重新转换为列表，则可以在结果连接上应用顺序。或者，您可以使用有序字典。

set1 = set(x[0] for x in arr1)
set2 = set(x[0] for x in arr2)
resultset = (set1 & set2)

这只会让您获得原始列表中日期的联合，为了重建 arr3 您需要在 arr1 和 arr2 中附加 [1:] 数据，其中日期位于结果集中。这种重建不会像使用上面的字典解决方案那样简洁，但是对于类似的问题，使用集合值得考虑。

于 2017-02-01T09:07:26.337 回答
2

您可以将数组转换为字典，然后再返回。

d1 = dict((x[0],x[1:]) for x in arr1)
d2 = dict((x[0],x[1:]) for x in arr2)
keys = set(d1).union(d2)
n = []
result = dict((k, d1.get(k, n) + d2.get(k, n)) for k in keys)
于 2013-07-16T17:13:14.140 回答
1

单字典方法：

tmp = {}
# add as many as you like into the outermost array.
for outer in [arr1,arr2]:
    for inner in outer:
        start, rest = inner[0], inner[1:]
        # the list if key exists, else create a new list. Append to the result
        tmp[start] = tmp.get(start,[]) + rest

output = []

for k,v in tmp.iteritems():
   output.append([k] + v)

这相当于完​​全外连接（即使一侧为空，也从两侧返回数据）。如果你想要一个内部连接，你可以把它改成这样：

tmp = {}
keys_with_dupes = []

for outer in [arr1,arr2]:
    for inner in outer:
        start, rest = inner[0], inner[1:]
        original = tmp.get(start,[])
        tmp[start] = original + rest
        if original:
            keys_with_dupes.append(start)

output = []

for k in keys_with_dupes:
   v = tmp[k]
   output.append([k] + v)
于 2013-07-16T17:12:45.360 回答
1

生成器函数方法，跳过日期不匹配的相应元素：

import itertools
def gen(a1, a2):
    for x,y in itertools.izip(a1, a2):
        if x[0] == y[0]:
            ret = list(x)
            ret.extend(y[1:])
            yield ret
        else:
            continue

>>print list(gen(arr1, arr2))
[['2011-10-10', 1, 1, 3, 4]]

但是，是的，如果可能的话，以不同的方式组织您的数据。

于 2013-07-16T17:21:38.353 回答
0

除非两者都是非常大的列表，否则我会使用字典：

arr1 = [
  ['2011-10-10', 1, 1],
  ['2007-08-09', 5, 3]
]

arr2 = [
  ['2011-10-10', 3, 4],
  ['2007-09-05', 1, 1]
]

table_1 = dict((tup[0], tup[1:]) for tup in arr1)
table_2 = dict((tup[0], tup[1:]) for tup in arr2)
merged = {}
for key, value in table_1.items():
    other = table_2.get(key)
    if other:
        merged[key] = value + other

否则，对每个排序，然后以这种方式进行合并会更有效。但我想对于大多数目的来说这已经足够快了。

于 2013-07-16T17:12:27.527 回答