DjangoBB LoFi version

Полная версия: (?) Работа с массивами в NumPy.

Начало » Python для новичков » (?) Работа с массивами в NumPy.

janitor

Ноя. 11, 2009 19:33:35

Необходимо дополнить все строки матрицы единицами

import numpy
arr=numpy.array([[-4,-4, 4],\
                             [-4, 4, 4],\
                             [ 4, 4, 4],\
                             [ 4,-4, 4],\
                             [-4,-4,-4],\
                             [-4, 4,-4],\
                             [ 4, 4,-4],\
                             [ 4,-4,-4]],"i")

print arr[0]
print numpy.append(arr[0],1)
#arr[0]=numpy.append(arr[0],1)
print arr[0]

#for x in arr:
     #y=append(x,1)
##

т.е. надо получить

arr=numpy.array([[-4,-4, 4, 1],\
                             [-4, 4, 4, 1],\
                             [ 4, 4, 4, 1],\
                             [ 4,-4, 4, 1],\
                             [-4,-4,-4, 1],\
                             [-4, 4,-4, 1],\
                             [ 4, 4,-4, 1],\
                             [ 4,-4,-4, 1]],"i")

numpy.append клеит все в одну строку, больше ничего не нашел. Если кто знает решение и/или достойную русскоязычную литературу по numpy, буду благодарен за помощь

asv13

Ноя. 11, 2009 20:24:33

Советую посмотреть примеры http://www.scipy.org/Numpy_Example_List Многое там и без английского понятно по выводимым результатам

>>> append(arr, [[1]]*8, axis = 1)
array([[-4, -4,  4,  1],
       [-4,  4,  4,  1],
       [ 4,  4,  4,  1],
       [ 4, -4,  4,  1],
       [-4, -4, -4,  1],
       [-4,  4, -4,  1],
       [ 4,  4, -4,  1],
       [ 4, -4, -4,  1]])