Как вставить дополнительный нужный столбец и строку?

у меня есть матрица

any_matrix = np.array( [[1,     2,   3,    4   ],
                        [5,     6,   7,    8   ],
                        [9,    10,   11,  12   ],
                        [13,   14,   15,   16  ]])

и я хочу вставить нужный столбец и строку между ними, как это (показано ниже)

  
[[1,   2,   0,   3,    4,  0   ],
[5,    6,   0,   7,    8,  0   ],
[0,    0,   1,   0,    0,  0   ],
[9,    10,  0,   11,  12,  0   ],
[13,   14,  0,   15,  16,  0   ],
[0,    0,   0,   0,    0,  1   ]]

моя попытка

import numpy as np


any_matrix = np.array( [[1,     2,   3,    4   ],
                        [5,     6,   7,    8   ],
                        [9,    10,   11,  12   ],
                        [13,   14,   15,   16  ]])


desired_cols_ids = np.array([2, 4], dtype=np.int64)
zero_arr_row = np.zeros((1, any_matrix .shape[1]))
any_matrix  = np.insert(any_matrix , desired_cols_ids, zero_arr_row, axis = 0)
zero_arr_col = np.array([0,0,1,0,0,0]).reshape(6,1)
any_matrix  = np.insert(any_matrix , desired_cols_ids, zero_arr_col, axis=1)

print(any_matrix)

>>output

[[ 1  2  0  3  4  0]
 [ 5  6  0  7  8  0]
 [ 0  0  1  0  0  1]
 [ 9 10  0 11 12  0]
 [13 14  0 15 16  0]
 [ 0  0  0  0  0  0]]

Попробуйте использовать встроенные hstack и vstack: Ссылка: Добавление строк и столбцов в массив numpy

Anshumaan Mishra 23.04.2022 09:30
Анализ настроения постов в Twitter с помощью Python, Tweepy и Flair
Анализ настроения постов в Twitter с помощью Python, Tweepy и Flair
Анализ настроения текстовых сообщений может быть настолько сложным или простым, насколько вы его сделаете. Как и в любом ML-проекте, вы можете выбрать...
7 лайфхаков для начинающих Python-программистов
7 лайфхаков для начинающих Python-программистов
В этой статье мы расскажем о хитростях и советах по Python, которые должны быть известны разработчику Python.
Установка Apache Cassandra на Mac OS
Установка Apache Cassandra на Mac OS
Это краткое руководство по установке Apache Cassandra.
Сертификатная программа "Кванты Python": Бэктестер ансамблевых методов на основе ООП
Сертификатная программа "Кванты Python": Бэктестер ансамблевых методов на основе ООП
В одном из недавних постов я рассказал о том, как я использую навыки количественных исследований, которые я совершенствую в рамках программы TPQ...
Создание персонального файлового хранилища
Создание персонального файлового хранилища
Вы когда-нибудь хотели поделиться с кем-то файлом, но он содержал конфиденциальную информацию? Многие думают, что электронная почта безопасна, но это...
Создание приборной панели для анализа данных на GCP - часть I
Создание приборной панели для анализа данных на GCP - часть I
Недавно я столкнулся с интересной бизнес-задачей - визуализацией сбоев в цепочке поставок лекарств, которую могут просматривать врачи и...
1
1
29
1
Перейти к ответу Данный вопрос помечен как решенный

Ответы 1

Ответ принят как подходящий

мы должны использовать как np.insert, так и np.hstack для этой операции

выглядит немного громоздко, но это будет работать !!

import numpy as np

any_matrix = np.array( [[1,     2,   3,    4   ],
                        [5,     6,   7,    8   ],
                        [9,    10,   11,  12   ],
                        [13,   14,   15,   16  ]])


desired_cols_ids = np.array([2, 4], dtype=np.int64)
zero_arr_row = np.zeros((1, any_matrix .shape[1]))
any_matrix  = np.insert(any_matrix , desired_cols_ids, zero_arr_row, axis = 0)
print(any_matrix)

# Array to be added as column at the last
column_to_be_added = np.array([0,0,0,0,0,1])
 
# Adding last column to numpy array
any_matrix = np.hstack((any_matrix, np.atleast_2d(column_to_be_added).T))

# Adding column at second position to numpy array
desired_cols_ids_1 = np.array([2,], dtype=np.int64)
zero_arr_col = np.array([0,0,1,0,0,0]).reshape(6,1)
any_matrix = np.insert(any_matrix, desired_cols_ids_1, zero_arr_col, axis=1)                   
                   

print(any_matrix)


Другие вопросы по теме