python - 如何根据 Numpy 数组的内容替换其特定条目

Question

假设我有一个由 ndarray 组成的简单矩阵(只是部分数据的示例):

import numpy as np
a = np.asarray([['1.0', 'Miami'],
   ['2.0', 'Boston'],
   ['1.4', 'Miami']]) 

我想在这个复杂的数据集中进行数据分析;) - 我想将“迈阿密”转换为 0，将波士顿转换为 1，以便使用真正奇特的 ML 算法。在 Python 中实现这一点的好方法是什么？ (我并不是要求明显的迭代和使用字典/if 语句来替换条目)，但如果有更好的方法使用 Numpy 或 native Python 来做到这一点，我会要求更多。

Answer 1

pandas是一个很好的工具。
首先将数组转换为 DataFrame:

In [11]: import pandas as pd

In [12]: df = pd.DataFrame(a, columns=['value', 'city'])

然后replace城市栏条目:

In [13]: df.city = df.city.replace({'Miami': 0, 'Boston': 1})

In [14]: df
Out[14]:
  value city
0   1.0    0
1   2.0    1
2   1.4    0