數組操作

Question

讓我們假設我有這個簡單的數組：

simple_list = [  
    ('1', 'a', 'aa'),  
    ('2', 'b', 'bb'),  
    ('3', 'c', 'cc') 
] 

如果我們考慮到這個列表作爲一個表，其中列由元組分開昏迷和線分離，我想創建一個函數只檢索我想要的列。例如，這個功能看起來像這樣的事情：

get_columns(array, tuple_columns_selector)) 

我想要的，例如，只收集了它的第一個和第三列，在這種情況下，它會返回我的另一個數組新值：

如果我這樣做：

get_columns(simple_list, (0,2))  
get_columns(simple_list, (0,)) 

它會返回類似：

[('1', 'aa'), ('2', 'bb'), ('1', 'cc')]  
[1, 2, 3] 

依此類推。你能幫我創建這個get_columns函數嗎？下面是我試過的代碼：

def get_columns(arr, columns): 
    result_list = [] 
    for ii in arr: 
     for i in columns: 
      result_list.append(ii[i]) 
    return result_list 


to_do_list = [ 
    ('Wake Up', True), 
    ('Brush Teeh', True), 
    ('Go to work', True), 
    ('Take a shower', True), 
    ('Go to bed', False) 
] 

print(get_columns(to_do_list, (0,))) 

Answer 1

使用的operator.itemgetter和map魔法：

from operator import itemgetter 

simple_list = [ 
    ('1', 'a', 'aa'), 
    ('2', 'b', 'bb'), 
    ('3', 'c', 'cc') 
] 

cols = (1,) # can be (0, 2) 
fn = itemgetter(*cols) 
print map(fn, simple_list) 

返回：

[('1', 'aa'), ('2', 'bb'), ('3', 'cc')] 

時cols是(0, 2)。

，並返回：

[1,2,3] 

時cols是(1,)。

所以你get_columns功能可以

def get_columns(data, cols): 
    return map(itemgetter(*cols), data) 

Answer 2

@kopos的答案看起來不錯，我只是想分享一個無需額外的庫。

simple_list = [ 
    ('1', 'a', 'aa'), 
    ('2', 'b', 'bb'), 
    ('3', 'c', 'cc') 
] 

def get_columns(array, tuple_columns_selector): 
    return [tuple(elem[i] for i in tuple_columns_selector) for elem in array] 

def get_columns_multiple_lines(array, tuple_columns_selector): 
    # The only difference between the result of this version and the other is that this one returns a list of lists 
    # while the other returns a list of tuples 
    resulting_list = [] # Create the variable that will store the resulting list 
    for elem in array: # Loop for each element in array 
     resulting_list.append([]) # We add a new "empty row" to store all the columns needed 
     for i in tuple_columns_selector: # Loop for each column needed 
      resulting_list[-1].append(elem[i]) # We append the column value to the last item in resulting_list 
    return resulting_list 


print get_columns(simple_list, (0,2)) # outputs [('1', 'aa'), ('2', 'bb'), ('3', 'cc')] 
print get_columns(simple_list, (0,)) # outputs [('1',), ('2',), ('3',)] 
print get_columns_multiple_lines(simple_list, (0,2)) # outputs [['1', 'aa'], ['2', 'bb'], ['3', 'cc']] 

唯一的區別是tuple_columns_selector只有一列時的返回值。如果這是一個重要的區別，我可以「糾正」它，但是你應該考慮如何使用這個價值，以及它是否適合具有不同的可能結構。