Доступ к столбцу матрицы без numpy

Question

Я пытаюсь написать программу для умножения матриц без использования numpy.

columns = [[], [], []]

for i in range(0, len(matrix)):
    columns[0].append(matrix[i][0])
    columns[1].append(matrix[i][1])
    columns[2].append(matrix[i][2])

Код, который я имею, может получить столбцы матрицы, но только для матрицы 3x3, и если я захочу использовать, скажем, матрицу 2x2, код не будет работать, если я не укажу для этого случая. Есть ли способ доступа к i-му столбцу матрицы в al oop, чтобы к столбцам матрицы можно было обращаться без предварительного знания размера?

Answer 1

Получите также другой размер из matrix:

for i in range(0, len(matrix)):
    for j in range(0, len(matrix[i])):
        columns[j].append(matrix[i][j])

Вы можете инициализировать columns следующим образом:

columns = [[] for x in matrix[0]]

(предполагается, что каждая строка имеет одинаковое число столбцов как первый).