如何在Python中合并两个不同维的numpy数组?

时间:2019-04-30 22:52:10

标签: python arrays numpy append concatenation

Python 2.7:

尝试尝试:

从1维Numpy数组添加具有datetype64(D)的列(arr_date) 现有的多维Numpy数组(数据)

出现以下错误:

  1. “ TypeError:无效的类型升级”
  2. 'numpy.AxisError:轴1超出维数组1的范围'

已创建的列,希望将其附加:

>> arr_date
<<     
[['2019-04-21']
 ['2019-04-21']
 ['2019-04-21']]

试图从源(数据)提供的3列中创建一个新的Numpy数组(arr_date)中的日期时间对象,然后使用以下方法将其添加到旧数组(数据)中:

  1. np.c _
  2. np.append
  3. np.hstack
  4. np.column_stack
  5. np.concatenate

data = [(2019, 4, 21, 4.9, -16.5447, -177.1961,  22.4, 'US')
(2019, 4, 21, 4.8,  -9.5526,  109.6003,  10. , 'UK')
(2019, 4, 21, 4.6,  -7.2737,  124.0192, 554.9, 'FR')]

arr_date = np.zeros((len(data),1), dtype='datetime64[D]')

i = 0

while i < len(data):    
        date = dt.date(data [i][0], data[i][1], data[i][2])     
        arr_date[i][0] = date    
        i += 1    


test1 = np.column_stack((data,arr_date))

np.c_[data, np.zeros(len(data))]

test2 = np.concatenate(data.reshape(-1,1), arr_date.reshape(-1,1), axis=1)

np.append(data, arr_date, axis = 1)

np.stack((data, arr_date), axis=-1)

np.hstack((data, arr_date))

test3 = np.column_stack((data, arr_date))

1 个答案:

答案 0 :(得分:1)

在您回答有关data.dtype的问题之前,我将添加逗号并将data列为元组列表:

In [117]: data = [(2019, 4, 21, 4.9, -16.5447, -177.1961,  22.4, 'US'), 
     ...: (2019, 4, 21, 4.8,  -9.5526,  109.6003,  10. , 'UK'), 
     ...: (2019, 4, 21, 4.6,  -7.2737,  124.0192, 554.9, 'FR')]                      

In [118]: arr_date = np.zeros((len(data),1), dtype='datetime64[D]') 
     ...:  
     ...: i = 0 
     ...:  
     ...: while i < len(data):     
     ...:         date = dt.date(data [i][0], data[i][1], data[i][2])      
     ...:         arr_date[i][0] = date     
     ...:         i += 1     
     ...:                                                                            

In [119]: arr_date                                                                   
Out[119]: 
array([['2019-04-21'],
       ['2019-04-21'],
       ['2019-04-21']], dtype='datetime64[D]')

所以arr_date是一个datetime64[D] dtype的(3,1)数组。

===

我猜测您的data实际上是具有复合dtype的结构化数组。例如:

In [121]: data1 = np.array(data, dtype='i,i,i,f,f,f,f,U2')                           

In [122]: data1                                                                      
Out[122]: 
array([(2019, 4, 21, 4.9, -16.5447, -177.1961,  22.4, 'US'),
       (2019, 4, 21, 4.8,  -9.5526,  109.6003,  10. , 'UK'),
       (2019, 4, 21, 4.6,  -7.2737,  124.0192, 554.9, 'FR')],
      dtype=[('f0', '<i4'), ('f1', '<i4'), ('f2', '<i4'), ('f3', '<f4'), ('f4', '<f4'), ('f5', '<f4'), ('f6', '<f4'), ('f7', '<U2')])

In [123]: data1.shape                                                                
Out[123]: (3,)

In [124]: data1.dtype                                                                
Out[124]: dtype([('f0', '<i4'), ('f1', '<i4'), ('f2', '<i4'), ('f3', '<f4'), ('f4', '<f4'), ('f5', '<f4'), ('f6', '<f4'), ('f7', '<U2')])

您的date迭代与此一起工作。但是data1的字段(不是列)可以按名称访问:

In [127]: data1['f0']                                                                
Out[127]: array([2019, 2019, 2019], dtype=int32)

column_stack可以将(3,)数组与(3,1)联接以产生(3,2),但是:

In [130]: np.column_stack((data, arr_date))                                          
---------------------------------------------------------------------------
TypeError                                 Traceback (most recent call last)
<ipython-input-130-5c8e6a103474> in <module>
----> 1 np.column_stack((data, arr_date))

/usr/local/lib/python3.6/dist-packages/numpy/lib/shape_base.py in column_stack(tup)
    638             arr = array(arr, copy=False, subok=True, ndmin=2).T
    639         arrays.append(arr)
--> 640     return _nx.concatenate(arrays, 1)

TypeError: invalid type promotion

首先请注意,尝试执行concatenate时发生错误。我敢打赌,其他所有随机尝试都会产生类似的错误(如果它们超过了轴错误)。该错误告诉我们,无法将Out[124]中的复合dtype与datetime64的{​​{1}} dtype组合在一起。 arr_date不匹配,因此无法匹配。

基本上,这不是串联问题。您并不是要在2D数组中添加“列”,甚至没有尝试创建2D数组。 dtypes不是2d。是1天。您需要做的是向结构化数组中添加一个字段。

有一个功能模块,可以更轻松地处理结构化数组。

data

In [131]: import numpy.lib.recfunctions as rf 应该可以解决问题,但是使用起来可能有些棘手:

append_fields

这仍然是一维数组,但是还有一个字段,我称之为In [137]: rf.append_fields(data1, 'date', arr_date.ravel(), usemask=False) Out[137]: array([(2019, 4, 21, 4.9, -16.5447, -177.1961, 22.4, 'US', '2019-04-21'), (2019, 4, 21, 4.8, -9.5526, 109.6003, 10. , 'UK', '2019-04-21'), (2019, 4, 21, 4.6, -7.2737, 124.0192, 554.9, 'FR', '2019-04-21')], dtype=[('f0', '<i4'), ('f1', '<i4'), ('f2', '<i4'), ('f3', '<f4'), ('f4', '<f4'), ('f5', '<f4'), ('f6', '<f4'), ('f7', '<U2'), ('date', '<M8[D]')])

===

我的回答:

Add and access object-type field of a numpy structured array

我展示了如何使用两个数组中的字段构造一个新的结构化数组,从而使您了解date在做什么。