將數組拆分為數組列表

Question

如何通過分組變量拆分2D數組，並請返回數組列表（順序也很重要）。

為了顯示預期的結果，可以用

> (A = matrix(c("a", "b", "a", "c", "b", "d"), nr=3, byrow=TRUE)) # input
     [,1] [,2]
[1,] "a"  "b" 
[2,] "a"  "c" 
[3,] "b"  "d" 
> (split.data.frame(A, A[,1])) # output
$a
     [,1] [,2]
[1,] "a"  "b" 
[2,] "a"  "c" 

$b
     [,1] [,2]
[1,] "b"  "d" 

編輯：澄清一下：我想根據第一列中的唯一值將數組/矩陣A拆分為多個數組的列表。 即，分割A成一個陣列，其中所述第一柱具有a ，而另一個陣列，其中所述第一柱具有b 。

我已經嘗試過將Python等效於R“ split”功能，但這給出了三個數組

import numpy as np
import itertools
A = np.array([["a", "b"], ["a", "c"], ["b", "d"]])
b = a[:,0]

def split(x, f):
     return list(itertools.compress(x, f)), list(itertools.compress(x, (not i for i in f)))
split(A, b) 

([array(['a', 'b'], dtype='<U1'),
  array(['a', 'c'], dtype='<U1'),
  array(['b', 'd'], dtype='<U1')],
 [])

還有numpy.split ，使用np.split(A, b) ，但是需要整數。 我雖然可以使用如何在Python中將字符串轉換為整數？ 將字母轉換為整數，但是即使我傳遞整數，也不會按預期拆分

c = np.transpose(np.array([1,1,2]))
np.split(A, c) # returns 4 arrays

能做到嗎？ 謝謝

編輯：請注意，這是一個小例子，並且組的數量可能大於兩個，並且可能未訂購。

Answer 1

您可以使用熊貓：

import pandas as pd
import numpy as np

a = np.array([["a", "b"], ["a", "c"], ["b", "d"]])

listofdfs = {}
for n,g in pd.DataFrame(a).groupby(0):
    listofdfs[n] = g

listofdfs['a'].values

輸出：

array([['a', 'b'],
       ['a', 'c']], dtype=object)

和，

listofdfs['b'].values

輸出：

array([['b', 'd']], dtype=object)

或者，您可以使用itertools groupby：

import numpy as np
from itertools import groupby
l = [np.stack(list(g)) for k, g in groupby(a, lambda x: x[0])]

l[0]

輸出：

array([['a', 'b'],
       ['a', 'c']], dtype='<U1')

和，

l[1]

輸出：

array([['b', 'd']], dtype='<U1')

Answer 2

如果我理解您的問題，可以進行簡單的切片，如下所示：

a = np.array([["a", "b"], ["a", "c"], ["b", "d"]])

x,y=a[:2,:],a[2,:]

x
array([['a', 'b'],
       ['a', 'c']], dtype='<U1')

y
array(['b', 'd'], dtype='<U1')