Как извлечь первые x цифр из матрицы?

Question

Я хотел бы извлечь первые 5 цифр из заданной матрицы nxm.

Например, матрица 3x3 A

A = [[ 1.  2.  3. ]
     [ 4.  5.  6.]
     [ 8.  9.  10.]]

Первые 5 цифр будут равны 1,2,3,4,5, поэтому первый ряд и половина второго.

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

Answer 1

Измените форму (-1,), затем нарежьте до x:

import numpy as np
A = np.arange(9).reshape((3,3)) 
print (A)

x = 5
B = A.reshape((-1,))[:x]
print (B)

[[0 1 2]
 [3 4 5]
 [6 7 8]]
[0 1 2 3 4]