Lặp qua mảng NumPy

Nếu bạn làm việc với mảng NumPy 1D, lặp qua tất cả phần tử có thể đơn giản như:

for x in my_array :
    ...

Nếu bạn làm việc với mảng NumPy 2D, mọi thứ phức tạp hơn. Một mảng 2D được tạo thành từ nhiều mảng 1D. Để duyệt tường minh qua từng phần tử riêng lẻ của một mảng đa chiều, bạn sẽ cần cú pháp sau:

for x in np.nditer(my_array) :
    ...

Hai mảng NumPy mà bạn có thể nhận ra từ khóa học nhập môn đã có sẵn trong phiên Python của bạn: np_height, một mảng NumPy chứa chiều cao của các cầu thủ Major League Baseball, và np_baseball, một mảng NumPy 2D chứa cả chiều cao (cột thứ nhất) và cân nặng (cột thứ hai) của những cầu thủ đó.