假設(shè)我有一個(gè)像這樣的numpy數(shù)組(更大,每個(gè)日期的重復(fù)次數(shù)不同):data = np.array([ \ ["2011-01-01", 24, 554, 66], \ ["2011-01-01", 44, 524, 62], \ ["2011-01-04", 23, 454, 32], \ ["2011-01-04", 22, 45, 42], \ ["2011-01-04", 14, 364, 12] \])現(xiàn)在,我想按日期將列分組為一個(gè)更扁平的結(jié)構(gòu):[ ["2011-01-01", [[24, 554, 66], [44, 524, 62]]], ["2011-01-04", [[23, 454, 32], [22, 45, 42], [14, 364, 12]]] ]我確實(shí)知道如何通過(guò)遍歷數(shù)組和附加元素來(lái)做到這一點(diǎn),但是在我看來(lái),這是極其不可思議的。是否有一些內(nèi)置的numpy函數(shù)來(lái)執(zhí)行此操作,或者有一些針對(duì)此類任務(wù)的自定義單行代碼?
使用numpy將數(shù)組分組為嵌套結(jié)構(gòu)
桃花長(zhǎng)相依
2021-03-25 10:09:02