在 python 中将数据从向量移动到二维数组的更快方法？

Question

This is my current code:

#move data from vector, A, into 2D array, H
    for i in range(0,nlat):
        jmin = nheader+1+i*nlon
        jmax = nheader+(i+1)*nlon
        A = A[jmin:jmax+1]
        H[i] = A
        A = np.array(Alist)

I have a very long vector, and I am taking certain data from it and putting it into a 2D array of dimension nlat x nlon. This setup works but it is very time consuming. Any suggestion of how to speed it up or rewrite it so it doesn't take as long would be very helpful.

Answer 1

You can use numpy .

import numpy as np

vector = np.array(A).reshape(nlat, nlon)