用多个元素替换列表中的一个元素

Question

我正在尝试修改包含两个列表的列表。对于两个内部列表中的每一个，我都会执行一些操作并将它们“拆分”为新列表。
这是我正在尝试做的一个简单示例：

[['a', 'b'], ['c', 'd']]  -->  [['a'], ['b'], ['c', 'd']]

目前，我的算法将 ['a', 'b'] 传递给一个函数，该函数确定是否应将其拆分为 [['a'], ['b']]（例如，基于它们的相关性）。该函数返回[['a'], ['b']]，它告诉我['a', 'b'] 应该被拆分，或者返回['a', 'b']（原始列表），这表明它不应该被拆分。

目前我有这样的事情：

blist = [['a', 'b'], ['c', 'd']]   #big list
slist =  [['a'], ['b']]            #small list returned by function

nlist = [items for i in xrange(len(blist)) for items in (slist if i==0 else blist[i])]

这会产生[['a'], ['b'], 'c', 'd']，而不是所需的输出[['a'], ['b'], ['c', 'd']]，后者不会改变原始列表中的第二个列表。我理解为什么会发生这种情况——在这种情况下，我的第二个循环也适用于 blist[1]，但我不确定如何解决它，因为我不完全理解列表理解。

首选“pythonic”解决方案。 任何反馈将不胜感激，谢谢！

编辑：正如标题所示，我正在尝试用['a'], ['b'] '替换'['a', 'b']。所以我希望'位置'是相同的，['a'], ['b'] 出现在['c', 'd'] 之前的原始列表中

结果 感谢 Christian、Paul 和 schwobaseggl 的解决方案！他们都工作:)

Answer 1

试试

...  else [blist[i]])]

创建列表列表。

Answer 2

你可以使用切片赋值：

>> l1 = [[1, 2], [3, 4]]
>>> l2 = [[1], [2]]
>>> l1[0:1] = l2
>>> l1
[[1], [2], [3, 4]]

这改变了l1，所以如果你想保留它，请先复制一份。

另一种不会改变l1 的方式是加法：

>> l1 = [[1, 2], [3, 4]]
>>> l3 = l2 + l1[1:]
>>> l3
[[1], [2], [3, 4]]

Answer 3

假设您具有决定是否拆分项目的上述功能：

def munch(item):
    if item[0] == 'a': # split
        return [[item[0]], [item[1]]]
    return [item] # don't split

您可以在简单的 for 循环中使用它。

nlist = []
for item in blist:
    nlist.extend(munch(item))

“Pythonic”是任何易于阅读和理解的东西。不要仅仅因为可以就使用列表推导式。

Answer 4

您可以更改拆分函数以返回结构上足够的列表。然后你可以使用理解：

def split_or_not(l):
  if condition: # split
    return [l[:1], l[1:]]
  return [l]  # wrap in extra list

# using map
nlist = [x for sub_l in map(split_or_not, blist) for x in sub_l]
# or nested comprehension
nlist = [x for sub_l in (split_or_not(l) for l in blist) for x in sub_l]