您的位置:

np.array添加元素

一、arraylist添加元素

  lst = [1, 2, 3, 4]
  print(lst)
  lst.append(5)
  print(lst)

在Python中,我们可以使用列表(list)的append()函数很容易地向列表添加元素。但是,在NumPy中,我们需要使用NumPy数组(array)来存储和处理数据,因此,我们需要对数组进行类似于列表的添加操作。

二、nparray增加元素

  import numpy as np
  arr = np.array([1, 2, 3, 4])
  print(arr)
  arr = np.append(arr, 5)
  print(arr)

np.append()函数可以使用来将元素添加到NumPy数组中。它需要两个参数,第一个是要添加元素的数组,第二个是要添加的元素。在这个例子中,我们通过在末尾添加5将元素添加到数组arr中。

三、np.concatenate连接数组

  import numpy as np
  arr1 = np.array([1, 2, 3])
  arr2 = np.array([4, 5, 6])
  print(arr1)
  print(arr2)
  arr3 = np.concatenate((arr1, arr2))
  print(arr3)

有时候,我们需要将两个数组连接在一起而不是添加单个元素。在NumPy中,我们可以使用np.concatenate()函数连接数组。它需要一个包含要连接的数组的元组作为参数。在这个例子中,我们将arr1和arr2连接成一个arr3数组。

四、np.insert插入元素

  import numpy as np
  arr = np.array([1, 2, 3, 4])
  print(arr)
  arr = np.insert(arr, 2, 5)
  print(arr)

除了添加元素到末尾以外,我们还可以使用np.insert()函数在NumPy数组中插入元素。它需要三个参数,第一个是要插入元素的数组,第二个是插入位置的索引,第三个是要插入的元素。在这个例子中,我们将5插入到索引为2的位置。

五、np.vstack和np.hstack连接数组

  import numpy as np
  arr1 = np.array([1, 2, 3])
  arr2 = np.array([4, 5, 6])
  print(arr1)
  print(arr2)
  arr3 = np.vstack((arr1, arr2))
  arr4 = np.hstack((arr1, arr2))
  print(arr3)
  print(arr4)

如果我们需要将两个数组水平或垂直堆叠在一起,我们可以使用np.vstack()和np.hstack()函数。np.vstack()函数将数组沿垂直方向堆叠,np.hstack()函数将数组沿水平方向堆叠。在这个例子中,我们将arr1和arr2垂直堆叠成arr3,并将它们水平堆叠成arr4。