python - 我们如何将 numpy 数组的两列连接成一列，并用空格分隔值？

Question

我有一个 numpy 字符串数据数组，我目前正在使用 data_subset = original_data[:, [1, 3, 8]] 提取它的一个子集。但是，我希望我的数据子集中的其中一列是我的原始数据的两列组合，并用一个空格连接。

我正在考虑的组合示例如下。我有 2 列代表名字和姓氏。作为示例数据，一行在第 3 列中有 John，在第 4 列中有 Smith，但在我的新数据中，我想要一个列 John Smith。是否有一个很好的 numpy 函数来执行这样的连接？

Answer 1

我建议在这里使用 pandas 库而不是 numpy —— 对字符串使用 numpy 数组通常麻烦多于它的值(value). OTOH，你想要的在 pandas 中非常简单:

>>> from pandas import DataFrame
>>> df = DataFrame({"first": ["John", "Jane"], "last": ["Smith", "Jones"]})
>>> df
  first   last
0  John  Smith
1  Jane  Jones
>>> df["first"] + " " + df["last"]
0    John Smith
1    Jane Jones

如果你绝对想使用 numpy，但是，如果你将 dtype 更改为 object，你可以做你想做的:

>>> import numpy as np
>>> a = np.array([["John", "Smith"], ["Jane", "Jones"]])
>>> a = a.astype(object)
>>> a[:,0] += " " + a[:,1]
>>> a = a[:,:1]
>>> a
array([[John Smith],
       [Jane Jones]], dtype=object)