在熊貓數據框和平均數組中按列分組

我有一個電影數據框，其中包含電影名稱，它們各自的類型和矢量表示（numpy數組）。在熊貓數據框和平均數組中按列分組

ID Year Title Genre Word Vector 
1 2003.0 Dinosaur Planet Documentary [-0.55423898, -0.72544044, 0.33189204, -0.1720... 
2 2004.0 Isle of Man TT 2004 Review Sports & Fitness [-0.373265237, -1.07549703, -0.469254494, -0.4... 
3 1997.0 Character Foreign [-1.57682264, -0.91265768, 2.43038678, -0.2114... 
4 1994.0 Paula Abdul's Get Up & Dance Sports & Fitness [0.3096168, -0.57186663, 0.39008939, 0.2868615... 
5 2004.0 The Rise and Fall of ECW Sports & Fitness [0.17175879, -2.38005066, -0.45771399, 1.32608...

我想組按流派，並得到各流派的平均向量表示（在體裁每部電影矢量分量明智的平均值）。

我第一次嘗試：

movie_df.groupby(['Genre']).mean()

但內置的均值函數不能採取numpy的陣列的平均值。

我試圖創造我自己的函數來進行，然後將其應用到各組，但我不知道這是用正確適用：

def vector_average(group): 
    series_to_array = np.array(group.tolist()) 
    return np.mean(series_to_array, axis = 0) 

movie_df.groupby(['Genre']).apply(vector_average)

任何指針將不勝感激！

來源

2017-08-17 Matt

可否請您打印出'df.head（5）'和它貼在這裏？ –

是的，但在問題中。 –

我不熟悉提供數據框樣本的最佳方式 - 此處的建議也將不勝感激！ – Matt

如果我理解正確的話，讓你可以簡單地應用np.mean到'Word Vector' SeriesGroupBy明確的逐分量的平均值。

df.groupby('Genre')['Word Vector'].apply(np.mean)

演示

>>> df = pd.DataFrame({'Title': list('ABCDEFGHIJ'), 
         'Genre': list('ABCBBDCDED'), 
         'Word Vector': [np.random.randint(0, 10, 10) 
             for _ in range(len('ABCDEFGHIJ'))]}) 

>>> df 

    Genre Title      Word Vector 
0  A  A [3, 6, 8, 0, 4, 8, 1, 4, 0, 1] 
1  B  B [5, 4, 4, 4, 8, 7, 4, 3, 7, 2] 
2  C  C [1, 7, 6, 7, 3, 3, 8, 1, 8, 1] 
3  B  D [0, 4, 6, 7, 1, 5, 5, 0, 6, 7] 
4  B  E [8, 2, 1, 4, 1, 2, 0, 4, 9, 1] 
5  D  F [7, 9, 7, 8, 8, 7, 2, 9, 1, 3] 
6  C  G [0, 7, 1, 9, 6, 2, 1, 0, 3, 7] 
7  D  H [4, 7, 9, 4, 1, 5, 0, 3, 0, 6] 
8  E  I [5, 1, 5, 1, 8, 1, 1, 4, 5, 6] 
9  D  J [7, 9, 0, 1, 8, 3, 8, 8, 1, 0] 

>>> df.groupby('Genre')['Word Vector'].apply(np.mean) 

Genre 
A [3.0, 6.0, 8.0, 0.0, 4.0, 8.0, 1.0, 4.0, 0.0, ... 
B [4.33333333333, 3.33333333333, 3.66666666667, ... 
C [0.5, 7.0, 3.5, 8.0, 4.5, 2.5, 4.5, 0.5, 5.5, ... 
D [6.0, 8.33333333333, 5.33333333333, 4.33333333... 
E [5.0, 1.0, 5.0, 1.0, 8.0, 1.0, 1.0, 4.0, 5.0, ... 
Name: Word Vector, dtype: object

來源

2017-08-17 04:29:47 miradulo

謝謝你這個作品！爲了完整起見，我嘗試了movie_df.groupby（['Genre']）。apply（np.mean）。它具有ID和年份列的輸出，但沒有列入矢量列？ – Matt

@perennial_nomad如果您嘗試在整個DataFrame上調用'np.mean'，它將僅爲具有數字數據類型的列提供結果 - 在這裏，''Word Vector''是類型對象。不客氣！ – miradulo

另一個後續 - 這會返回一個熊貓系列，當我嘗試使用to_frame將其寫入數據框時，它只返回帶有流派作爲標籤的Word向量列。有沒有一種方法可以直接轉換爲20×2的DF和'流派'和'字矢量'？ – Matt

在熊貓數據框和平均數組中按列分組

回答

相關問題