Блог

В приведенном ниже коде я пытался использовать np.append , но он дает мне только последнее значение, добавленное в цикле.

У меня есть два вложенных цикла for (один выполняется с индексом i , другой с k ). Теперь я хочу использовать определенную функцию для разных индексов, и я хочу, чтобы результаты были сведены в один массив. Таким образом, чтобы каждая строка в массиве содержала результат для одного индекса.

Вот минимальный воспроизводимый пример:

 import numpy as np

a = np.arange(0,2,1)
b = np.arange(3,5,1)
c = np.array([])

for i in range(0,2,1):
    for k in range(0,2,1):
        c = np.append(a[i],b[k])
print(c)
  

Результат таков [1 4] . Но я хочу, чтобы один единственный вектор c содержал ([0 3][0 4][1 3][1 4])

Вы создаете 1d массив вместо 2d. и np.append вам нужно передать текущий массив и добавляющие элементы, и он вернет новый массив.

 a = np.arange(0,2,1)
b = np.arange(3,5,1)
c = np.empty(shape=[0, 2])

for i in range(0,2,1):
    for k in range(0,2,1):
        c = np.append(c,[[a[i],b[k]]],axis=0)
        
print(c)
  

Вывод:

 [[0. 3.]
 [0. 4.]
 [1. 3.]
 [1. 4.]]
  

Вас смущает синтаксис для np.append()

Используйте ниже для правильной реализации:

 c = np.append(c,[a[i],b[k]])
  

Также для этого вы получите вывод, напечатанный как:

 [0. 3. 0. 4. 1. 3. 1. 4.]
  

Я запутался в векторе, о котором вы говорите : ([0 3][0 4][1 3][1 4])

Указанное вами представление не является вектором, а выглядит как матрица с размерностью 4x2 , где .reshape() может вам помочь.

Если вы этого хотите, просто напишите: c.reshape((4,2)) что даст вам:

 [[0. 3.]
 [0. 4.]
 [1. 3.]
 [1. 4.]]