python - 从数据列表中创建矩阵列表

Question

我有一个来自 .txt 文档的假设数据列表

0 1 2 3 4 5 6 7 8 (值)

我想制作 2 个这样的矩阵列表

list 1            list 2
[0, 1, 2]         [[0],[1],[2]]
[3, 4, 5]         [[3],[4],[5]]
[6, 7, 8]         [[6],[7],[8]]

列表 1 中的每一项与先前声明的矩阵相乘时都必须能够给出点积。

new list 1
dot([0,1,2],mat)
dot([3,4,5],mat)
dot([6,7,8],mat)

我尝试将 .txt 文档中的 3 个值作为矩阵附加到列表中(这是在 for 循环中完成的)

list1.append ([value[0], value[1], value[2]]

但是没有成功。它给了我一个错误。

感谢任何帮助。

Answer 1

您可以通过列表理解来实现它:

>>> my_string = '0 1 2 3 4 5 6 7 8'
>>> row = 3
>>> my_list = my_string.split() # split the string based on space ' '

>>> list_1 = [my_list[i*3:(i*3)+row] for i in range(row)]
>>> list_1 # Output for example 1
[['0', '1', '2'], 
 ['3', '4', '5'], 
 ['6', '7', '8']]

>>> list_2 = [[[c] for c in my_list[i*3:(i*3)+row]] for i in range(row)]
>>> list_2 # Ouput for example 2
[[['0'], ['1'], ['2']], 
 [['3'], ['4'], ['5']], 
 [['6'], ['7'], ['8']]]