What’s the simplest way to extend a numpy array in 2 dimensions?

by

The shortest in terms of lines of code i can think of is for the first question.

>>> import numpy as np
>>> p = np.array([[1,2],[3,4]])

>>> p = np.append(p, [[5,6]], 0)
>>> p = np.append(p, [[7],[8],[9]],1)

>>> p
array([[1, 2, 7],
   [3, 4, 8],
   [5, 6, 9]])

And the for the second question

    p = np.array(range(20))
>>> p.shape = (4,5)
>>> p
array([[ 0,  1,  2,  3,  4],
       [ 5,  6,  7,  8,  9],
       [10, 11, 12, 13, 14],
       [15, 16, 17, 18, 19]])
>>> n = 2
>>> p = np.append(p[:n],p[n+1:],0)
>>> p = np.append(p[...,:n],p[...,n+1:],1)
>>> p
array([[ 0,  1,  3,  4],
       [ 5,  6,  8,  9],
       [15, 16, 18, 19]])

More Related Contents:

  1. What are the advantages of NumPy over regular Python lists?
  2. ValueError: setting an array element with a sequence
  3. Numpy `logical_or` for more than two arguments
  4. What is the difference between contiguous and non-contiguous arrays?
  5. Quick way to upsample numpy array by nearest neighbor tiling [duplicate]
  6. How does NumPy’s transpose() method permute the axes of an array?
  7. Count consecutive occurences of values varying in length in a numpy array
  8. Concatenate two NumPy arrays vertically
  9. How to make scipy.interpolate give an extrapolated result beyond the input range?
  10. Getting the indices of several elements in a NumPy array at once
  11. How to do n-D distance and nearest neighbor calculations on numpy arrays
  12. Multiplying across in a numpy array
  13. initialize a numpy array
  14. How to conditionally combine two numpy arrays of the same shape
  15. What does dtype=object mean while creating a numpy array?
  16. Numpy: efficient way to generate combinations from given ranges
  17. Selecting close matches from one array based on another reference array
  18. Python: slicing a multi-dimensional array
  19. NumPy chained comparison with two predicates
  20. How to count values in a certain range in a Numpy array?
  21. Why does numpy have a corresponding function for many ndarray methods?
  22. Numpy array dimensions
  23. Efficiently sorting a numpy array in descending order?
  24. Numpy: Divide each row by a vector element
  25. How to remove all rows in a numpy.ndarray that contain non-numeric values
  26. Setting the fmt option in numpy.savetxt
  27. Access n-th dimension in python [duplicate]
  28. Efficient dot products of large memory-mapped arrays
  29. Mapping a NumPy array in place
  30. numpy array row major and column major

Leave a Comment