從包含數組的數據框列中提取數組

Question

我正在用荊棘和耳蝸產生刺突。 結果采用pandas.DataFrame的形式，其中“ spikes”列包含我需要的穗時間數組。

荊棘生成的DataFrame

當我嘗試用

a = df.as_matrix(['spikes'])

結果矩陣將始終如下所示：

a = [[array([1,2,3])][array([4,5,6])]]

為了進一步使用數據，我需要表格：

a= [[1,2,3][4,5,6]]

如何刪除array（）部分？

這是我使用的代碼：

import cochlea

import thorns.waves as wv

#generating the spiketrain

sound = wv.ramped_tone(fs=100e3,freq=500,duration=0.3,dbspl=60)
spiketrain=cochlea.run_zilany2014(sound,fs=100e3,anf_num=(2,0,0),cf=500,seed=0,species='human')
print spiketrain

#trying to extrakt spiketime information
a = spiketrain.as_matrix(['spikes'])
print 'a='
print a

Answer 1

這是一種方法：

import pandas as pd, numpy as np

df = pd.DataFrame({'A': [np.array([1,2,3]), np.array([4,5,6]), np.array([7,8,9])]})

res = np.array(df['A'].values.tolist())

# array([[1, 2, 3],
#        [4, 5, 6],
#        [7, 8, 9]])

res2 = list(map(list, df['A']))

# [[1, 2, 3],
#  [4, 5, 6],
#  [7, 8, 9]]