Mảng thao tác trong python

Question

Giả sử tôi có mảng đơn giản này:

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

Nếu chúng ta xem xét danh sách này như một bảng, nơi các cột được phân cách bằng hôn mê và dòng ngăn cách bởi các bộ, tôi muốn tạo một hàm chỉ lấy các cột mà tôi muốn. ví dụ, chức năng này sẽ trông giống như một cái gì đó như thế này:

get_columns(array, tuple_columns_selector)) 

tôi muốn, ví dụ, để thu thập chỉ cột đầu tiên và thứ ba ra khỏi nó, trong trường hợp này, nó sẽ trả lại cho tôi mảng khác với các mới giá trị:

nếu tôi làm:

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

nó sẽ trở lại một cái gì đó như:

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

Và cứ thế. Bạn có thể giúp tôi tạo hàm get_columns này không? Dưới đây là đoạn code tôi đã cố gắng:

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

Sử dụng sự kỳ diệu của operator.itemgetter và 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) 

Returns:

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

khi cols là (0, 2).

Và nó sẽ trả về:

[1,2,3] 

khi cols là (1,).

Vì vậy, chức năng get_columns của bạn có thể

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

Answer 2

Câu trả lời của @kopos có vẻ tốt đẹp, tôi chỉ muốn chia sẻ một mà không có thư viện bổ sung.

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']] 

Sự khác biệt duy nhất là giá trị trả về khi tuple_columns_selector chỉ là một cột. Nếu đó là một sự khác biệt quan trọng, tôi có thể "sửa" nó, nhưng bạn nên suy nghĩ về cách giá trị đó sẽ được sử dụng và nếu nó thuận tiện cho nó để có cấu trúc khác nhau có thể.