python - 如何在 Python 中编码分类值

Question

给定词汇表["NY", "LA", "GA"]，如何以这样一种方式对其进行编码，使其成为:

"NY" = 100
"LA" = 010
"GA" = 001

所以如果我在 "NY GA" 上查找，我会得到 101

Answer 1

你可以使用numpy.in1d:

>>> xs = np.array(["NY", "LA", "GA"])
>>> ''.join('1' if f else '0' for f in np.in1d(xs, 'NY GA'.split(' ')))
'101'

或:

>>> ''.join(np.where(np.in1d(xs, 'NY GA'.split(' ')), '1', '0'))
'101'