numpy.ndarray 实现对特定行或列取值

以下是numpy.ndarray实现对特定行或列取值的攻略:

numpy.ndarray实现对特定行或列取值

NumPy中,可以使用切片和索引来实现对特定行或列取值。以下是一些示例:

对特定行取值

可以使用切片来对特定行取值。以下是一个示例:

import numpy as np

a = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
row_1 = a[0, :]
print(row_1)

输出:

[1 2 3]

对特定列取值

可以使用切片来对特定列取值。以下是一个示例:

import numpy as np

a = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
col_1 = a[:, 0]
print(col_1)

输出:

[1 4 7]

对特定行和列取值

可以使用切片来对特定行和列取值。以下是一个示例:

import numpy as np

a = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
sub_array = a[0:2, 0:2]
print(sub_array)

输出:

[[1 2]
 [4 5]]

总结

这就是numpy.ndarray实现对特定行或列取值的攻略。可以使用切片和索引来实现对特定行或列取值,也可以使用切片来对特定行和列取值。希望这篇文章能够帮助您更好地理解numpy.ndarray实现对特定行或列取值的方法。

本文链接:http://task.lmcjl.com/news/16740.html

展开阅读全文