如何在不进行切片的情况下从数组中检索元素



假设我有一个数组A1。B1是存储A1的最后两行的另一个阵列。我需要检索A1的前两行,而不将其作为另一个数组(C1(进行切片。从逻辑上讲,我想的是A1(整个数组(-B1(最后两行(=C1(前两行(,但我不知道Python是否会这样做。如何在不进行切片的情况下从数组中获取元素?感谢你的帮助!

A1 = np.array([ [1, 4, 6, 8],[2, 5, 7, 10],[3, 6, 9, 13],[11, 12, 16, 0]])B1 = A1[2:4,]

您可以删除最后两行,而不是"减去"这两个数组:

C1 = np.delete(A1, (2,3), axis=0)
array([[ 1,  4,  6,  8],
[ 2,  5,  7, 10]])
def first_2_rows_of_A1():
for i,row in enumerate(A1):
yield row
if i >= 1:
break
first_2_rows = list(first_2_rows_of_A1())

我想这正是你想要的。。。但我不明白为什么你不把前两行作为A1[:2]

最新更新