在numpy中用逗号将元素从二维数组中分隔开，使字符串组成一维数组

Question

我需要从一个2D数组生成一个1D numpy数组，这样2列中的元素被连接和分离，数据类型是字符串。我可以用np.split来实现相反的功能，但是np.concatenate似乎没有我所需要的方式工作，也没有像pandas中那样的'join'函数。在

这些是我的数组，它们的形状是（53187L，2L）和（27530L，2L）：

dfSTcombined=np.vstack([txx_copy,dfSTA]).T locatsCruSta=np.vstack([locatsID[:,3],locatsID[:,4]]).T

当我试图比较它们的成员资格时，问题随之而来：

dfSTinlocats=np.in1d(locatsCruSta, dfSTcombined)
dfSTinlocats.shape
->(55060L, 2L)
这样得到的长度是我预期的两倍（55060L=2*27530），因为它分别比较数组中的两列，但我需要同时比较两列。在

尝试其他方法：
test4=np.where(dfSTcombined==locatsCruSta)
-> ValueError: shape mismatch: objects cannot be broadcast to a single shape

所以我需要创建一个新数组，其中1列包含两列的字符串，还有一个分隔符。在

[AR1006, 02]（2列）
[AR1024, 05]

应该变成：

['AR1006,02']（1列，尽管我的符号不正确）

最终目标是使用类似于numpy.in1D的方法在另一个组中找到一个组的成员，然后使用该成员资格的索引来选择更多的数据来匹配两个巨大的数据集。如果能在熊猫身上更容易些，我就不喜欢纽比了，但我不知道如何比较熊猫的会员资格。在

Answer 1

你提到熊猫是个选择。我当然会选择这个库，因为除此之外，它还有一些非常有用的函数来处理字符串（尽管我确信你想要的在NumPy中也是可能的）。在

>>> import pandas as pd
>>> import numpy as np

将你的NumPy数组放入Pandas数据帧：

^{pr2}$

可以很容易地将两列与分隔符一起连接到pd.Series对象中：

>>> series = df.a + ", " + df.b
>>> series

0    AR1006, 02
1    AR1007, 01
2    AR1008, 03
dtype: object

如果您愿意，您甚至可以使用np.array(series)将该系列转换回NumPy数组。在

检查序列对象的成员身份也很简单。实际上，您可以使用np.in1d：

>>> np.in1d(df.a, df.b)
array([False, False, False], dtype=bool)

>>> np.in1d(['02', '01'], df.b)
array([ True,  True], dtype=bool)