我有一个包含数组的numpy数组:
array([[1,2,3],[4,5,6],[7,8,9]])
我如何找到这些内部数组的所有最后元素的最大值?即。对于这种情况,返回值为9 (max(3,6,9))
我能够通过首先转换为Pandas数据帧来实现这一点,但这会减慢程序速度,我确信使用np会更简单。
答案 0 :(得分:1)
假设你打算写
array([[1,2,3],[4,5,6],[7,8,9]])
然后您可以使用A[:,-1]
对最后一列进行切片并调用.max()
。