Question

我正在尝试在python中加入recarrys，以便相同的值连接到许多元素。以下代码在1：1比例时有效，但当我尝试执行多项操作时：1，它只加入一个实例：

import numpy as np
import matplotlib
# First data structure
sex = np.array(['M', 'F', 'M', 'F', 'M', 'F'])
causes = np.array(['c1', 'c1', 'c2', 'c2', 'c3', 'c3'])
data1 = np.core.records.fromarrays([sex, causes], names='sex, causes')

# Second data structure
causes2 = np.array(['c1', 'c2', 'c3'])
analyst = np.array(['p1', 'p2', 'p3'])
data2 = np.core.records.fromarrays([causes2, analyst], names='causes, analyst')

# Join on Cause
all_data = matplotlib.mlab.rec_join(('causes'), data1, data2, jointype='leftouter')

我想要包含的all_data重新数组是data1中的所有数据，数据2中指示了相应的分析。

Answer 1

可能有一个很好的使用记录数组，但我认为python dict应该在这里一样好...想知道自己这样做的笨拙方式，如果它是好的。

dct = dict(zip(data2['causes'], data2['analyst']))
all_data = mlab.rec_append_fields(data1, 'analyst',
    [dct[x] for x in data1['causes']])

通过Python中的属性加入Recarrays

1 个答案: