python - 如何在 Python 中替换多维列表的第 n 个元素？

Question

我有一个这样的列表：

a = [['0', '0'], ['0', '0'], ['0', '0']]

我正在尝试编写一个以第 n 个列表元素为参数并将 0 替换为“X”的函数。例如：函数replace([1, 2])会做：

 a = [['0', 'X'], ['X', '0'], ['0', '0']]

换句话说，替换函数应该将 a 视为一个连续列表。

由于每个列表都是一个元素，我认为这是不可能的。这可能吗？

Answer 1

import math
def replace(a, t):
    for i in t:
        a[int(math.floor(i/2))][i % 2] = "X"

    return a

replace([['0', '0'], ['0', '0'], ['0', '0']], [1,2])
>> [['0', 'X'], ['X', '0'], ['0', '0']]

Answer 2

您的replace()功能没有意义，但是在您想要的任何列表中替换任何您想要的东西肯定是可能的而且确实很简单。

>>> a = [['0', '0'], ['0', '0'], ['0', '0']]
>>> a[0][1] = '1'
>>> a[1][0] = '2'
>>> a
[['0', '1'], ['2', '0'], ['0', '0']]
>>> 

Answer 3

鉴于：

a = [['0', '0'], ['0', '0'], ['0', '0']]

您可以展平列表：

>>> [e for sub in a for e in sub]
['0', '0', '0', '0', '0', '0']

然后元素线性映射：

>>> fl=[e for sub in a for e in sub]
>>> fl[1]=1
>>> fl[2]=2
>>> fl
['0', 1, 2, '0', '0', '0']

如果您展平列表，则可以使用切片分配：

>>> fl[1:2]='XX'
>>> fl
['0', 'X', 'X', '0', '0', '0', '0', '0', '0', '0', '0']

如果您愿意，然后重新组合子组：

>>> [list(e) for e in zip(*[fl[i::2] for i in range(2)])]
[['0', 'X'], ['X', '0'], ['0', '0']]

您还可以转换为多维下标：

li=[[1,2,3],[4,5],[6],[7,8,9,10]]

def xlate(li,wanted):
    idx=0
    for i,e in enumerate(li):
        for j,e_ in enumerate(e):
            if idx==wanted: return (i,j)
            idx+=1


    return (None,None)        

t=xlate(li,5)
li[t[0]][t[1]]='X'  

印刷：

[[1, 2, 3], [4, 5], ['X'], [7, 8, 9, 10]]