Извлечение определенных столбцов в массиве numpy

Question

Извлечение определенных столбцов в массиве numpy

Это простой вопрос, но скажите, что у меня есть матрица MxN. Все, что я хочу сделать, это извлечь определенные столбцы и сохранить их в другом массиве numpy, но я получаю недопустимые синтаксические ошибки.
Вот код:

extractedData = data[[:,1],[:,9]].

Кажется, что выше строки должно быть достаточно, но я думаю, что нет. Я огляделся, но не смог найти ничего синтаксически мудрого относительно этого конкретного сценария.

627 8

python numpy syntax

8 ответов:

Comments

Ничего не найдено.

Fred Foo · Accepted Answer · 2018-09-21 13:37:07

Я предполагаю, что вы хотели столбцы 1 и 9? Это
data[:, [1, 9]]
или с именами:
data[:, ['Column Name1','Column Name2']]
вы можете получить имена от data.dtype.names...

Michael J. Barber · Accepted Answer · 2011-12-05 16:26:02

предполагая, что вы хотите получить столбцы 1 и 9 с этим фрагментом кода, он должен быть:
extractedData = data[:,[1,9]]

queise · Accepted Answer · 2015-06-01 14:14:00

Если вы хотите извлечь только некоторые столбцы:
idx_IN_columns = [1, 9]
extractedData = data[:,idx_IN_columns]
Если вы хотите исключить определенные столбцы:
idx_OUT_columns = [1, 9]
idx_IN_columns = [i for i in xrange(np.shape(data)[1]) if i not in idx_OUT_columns]
extractedData = data[:,idx_IN_columns]

Daksh · Accepted Answer · 2018-03-05 18:20:51

Я хотел бы отметить, что если количество столбцов, которые вы хотите извлечь, равно 1 результирующей матрице не было бы матрицы Mx1 как и следовало ожидать, но вместо этого массив, содержащий элементы столбца, который вы извлекли.

преобразовать его в матрицу изменить форму(M, 1) метод должен использоваться в результирующем массиве.

Rahul · Accepted Answer · 2017-09-04 12:34:10

вы можете использовать extracted_data = данные.ix [:, ['Column1', 'Column2']]

Jan Kukacka · Accepted Answer · 2018-02-05 18:25:05

еще одна вещь, на которую вы должны обратить внимание при выборе столбцов из массива N-D, используя такой список:
data[:,:,[1,9]]
Если вы удаляете измерение (например, выбрав только одну строку),результирующий массив будет (по какой-то причине) перестановкой. Итак:
print data.shape            # gives [10,20,30]
selection = data[1,:,[1,9]]
print selection.shape       # gives [2,20] instead of [20,2]!!

yanhh · Accepted Answer · 2018-07-28 04:25:55

всего:

>>> m = np.matrix(np.random.random((5, 5)))
>>> m
matrix([[0.91074101, 0.65999332, 0.69774588, 0.007355  , 0.33025395],
        [0.11078742, 0.67463754, 0.43158254, 0.95367876, 0.85926405],
        [0.98665185, 0.86431513, 0.12153138, 0.73006437, 0.13404811],
        [0.24602225, 0.66139215, 0.08400288, 0.56769924, 0.47974697],
        [0.25345299, 0.76385882, 0.11002419, 0.2509888 , 0.06312359]])
>>> m[:,[1, 2]]
matrix([[0.65999332, 0.69774588],
        [0.67463754, 0.43158254],
        [0.86431513, 0.12153138],
        [0.66139215, 0.08400288],
        [0.76385882, 0.11002419]])

столбцы не должны быть в порядке:

>>> m[:,[2, 1, 3]]
matrix([[0.69774588, 0.65999332, 0.007355  ],
        [0.43158254, 0.67463754, 0.95367876],
        [0.12153138, 0.86431513, 0.73006437],
        [0.08400288, 0.66139215, 0.56769924],
        [0.11002419, 0.76385882, 0.2509888 ]])

Pranav Mahajan · Accepted Answer · 2017-08-02 20:18:14

вы также можете использовать extractedData=data([:,1],[:,9])