可以使用以下代码来检查argmax是否对字符串数组正常工作:
import numpy as np
# 创建字符串数组
arr = np.array(['apple', 'banana', 'cherry', 'date', 'elderberry'])
# 使用argmax找到最长字符串的索引
max_index = np.argmax([len(s) for s in arr])
# 打印最长字符串和其索引
print("最长的字符串:", arr[max_index])
print("索引:", max_index)
运行上述代码,将会输出:
最长的字符串: elderberry
索引: 4
说明argmax对字符串数组正常工作,找到了最长字符串的索引。