Как выбрать элементы построчно из массива NumPy? - PullRequest
Новая
       14

Как выбрать элементы построчно из массива NumPy?

10 голосов
/ 25 сентября 2011

У меня есть такой массив: 

dd= [[foo 0.567 0.611]
     [bar 0.469 0.479]
     [noo 0.220 0.269]
     [tar 0.480 0.508]
     [boo 0.324 0.324]]

Как бы один цикл проходил по массиву, выбирая foo и получая 0.567 0.611 с плавающей точкой в ​​виде синглтона?Затем выберите bar и получите 0.469 0.479 как float как синглтон .....

Я мог бы получить вектор первых элементов в виде списка, используя 

dv=  dd[:,1]

'foo', и 'Элементы bar не являются неизвестными переменными, они могут меняться.

Как бы я изменился, если бы элемент находился в позиции [1]? 

[[0.567 foo2 0.611]
  [0.469 bar2 0.479]
  [0.220 noo2 0.269]
  [0.480 tar2 0.508]
  [0.324 boo2 0.324]]

Ответы [ 2 ]

21 голосов
/ 25 сентября 2011

Вы поместили тег NumPy в свой Вопрос, поэтому я предполагаю, что вы хотите использовать синтаксис NumPy, который в ответе перед моим не используется.

Если на самом деле вы хотите использовать NumPy, то, скорее всего, вам не нужны строки в вашем массиве, в противном случае вам также придется представлять ваши плавающие числа как строки.

То, что вы ищете, это синтаксис NumPy для доступа к элементам двумерного массива по строкам (и исключая первый столбец) .

Этот синтаксис: 

M[row_index,1:]        # selects all but 1st col from row given by 'row_index'

W / r / t второй сценарий в вашем Вопросе - выбор несмежных столбцов : 

M[row_index,[0,2]]     # selects 1st & 3rd cols from row given by 'row_index'


Небольшое осложнение в вашем Вопросе состоит в том, что вы хотите использовать строку для row_index, поэтому необходимо удалить строки (чтобы вы могли создать массив чисел с плавающей точкой NumPy), заменив их числовой строкой индексы, а затем создать справочную таблицу для сопоставления строк с числовыми индексами строк : 

>>> import numpy as NP
>>> # create a look-up table so you can remove the strings from your python nested list,
>>> # which will allow you to represent your data as a 2D NumPy array with dtype=float
>>> keys
      ['foo', 'bar', 'noo', 'tar', 'boo']
>>> values    # 1D index array comprised of one float value for each unique string in 'keys'
      array([0., 1., 2., 3., 4.])
>>> LuT = dict(zip(keys, values))

>>> # add an index to data by inserting 'values' array as first column of the data matrix
>>> A = NP.hstack((vals, A))
>>> A
        NP.array([  [ 0., .567, .611],
                    [ 1., .469, .479],
                    [ 2., .22, .269],
                    [ 3., .48, .508],
                    [ 4., .324, .324] ])

>>> # so now to look up an item, by 'key':
>>> # write a small function to perform the look-ups:
>>> def select_row(key):
        return A[LuT[key],1:]

>>> select_row('foo')
      array([ 0.567,  0.611])

>>> select_row('noo')
      array([ 0.22 ,  0.269])

Второй сценарий в вашем Вопросе: что если столбец индекса изменится? 

>>> # e.g., move index to column 1 (as in your Q)
>>> A = NP.roll(A, 1, axis=1)
>>> A
      array([[ 0.611,  1.   ,  0.567],
             [ 0.479,  2.   ,  0.469],
             [ 0.269,  3.   ,  0.22 ],
             [ 0.508,  4.   ,  0.48 ],
             [ 0.324,  5.   ,  0.324]])

>>> # the original function is changed slightly, to select non-adjacent columns:
>>> def select_row2(key):
        return A[LuT[key],[0,2]]

>>> select_row2('foo')
        array([ 0.611,  0.567])
2 голосов
/ 25 сентября 2011

Во-первых, вектор первых элементов равен 

dv = dd[:,0]

(Python 0-indexed)

Во-вторых, для обхода массива (и, например, для хранения в формате dict) вы пишете: 

dc = {}
ind = 0 # this corresponds to the column with the names
for row in dd:
    dc[row[ind]] = row[1:]
...