Задача по матрицам

Выполнить обработку элементов прямоугольной матрицы A, имеющей N строк и M столбцов. Добавить к матрице столбец чисел и вставить его под номером K. Вот что есть на данный момент:

import numpy as np
print('введите размер матрицы в строку: ')
n, m = [int(i) for i in input().split()]
print("введите матрицу через enter")
a = np.array([[int(j) for j in input().split()] for i in range(n*m)]).reshape(n, m)
print("введите столбец через enter")
f = np.array([[int(z) for z in input().split()] for z in range(n)]).reshape(n, 1)
print("введите k")
k = int(input())
a = np.insert(a, k, f, axis = 1)
print(a)

проблема в том, что в ответе столбец F повторяется N раз.

Источник: https://ru.stackoverflow.com/questions/1267636/%D0%97%D0%B0%D0%B4%D0%B0%D1%87%D0%B0-%D0%BF%D0%BE-%D0%BC%D0%B0%D1%82%D1%80%D0%B8%D1%86%D0%B0%D0%BC