如何在Python中将0和1的数组转换为字符串数组？

Question

我有以下python数组：np.array([1,1,1,0,0])，我想根据条件if 1, 'yes', else 'no'将其转换为以下数组：np.array(['yes','yes','yes','no','no])。最好的办法是什么？我很乐意在原地或单独的阵列中进行。你知道吗

Answer 1

使用np.where：

import numpy as np

arr = np.array([1,1,1,0,0])

result = np.where(arr, 'yes', 'no')
print(result)

输出

['yes' 'yes' 'yes' 'no' 'no']

Answer 2

您可以使用列表理解：

import numpy as np

arr = np.array([1,1,1,0,0])

result = np.array(['yes' if x else 'no' for x in arr])

print(result)

输出：['yes' 'yes' 'yes' 'no' 'no']