我正在运行Python代码，并在keepdims上收到此错误：
enter image description here
这是代码：
enter image description here
几天前在我的电脑上运行这个命令效果很好，但我已经运行了其他代码等，之后可能做了一些事情。
在amax上写keepdims是可行的，但在argmax上不行。我朋友现在在她的电脑上运行了同样的程序，即使代码相同，这个错误也没有出现。我试着卸载并重新安装anaconda，但它没有改变它。不确定是否有其他东西需要下载或发生了什么。
1条答案
按热度按时间6kkfgxo01#
对于数组
x，复制np.argmax(x, axis=0, keepdims=True)行为的简单方法是np.argmax(x, axis=0)[np.newaxis, ...]。注意，这是专门针对axis=0的情况。其他替代项包括
np.expand_dims(np.argmax(x, axis=0), 0)和np.argmax(x, axis=0).reshape((1,) + x.shape[1:])。对于任意轴
k，np.expand_dims(np.argmax(x, axis=k), k)将起作用。