सरणी हेरफेर

Question

की मैं इस सरल सरणी है लगता है:

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

अगर हम एक मेज, जहां कॉलम comas और लाइनों tuples द्वारा अलग से अलग होते हैं के रूप में इस सूची पर विचार, मैं एक समारोह का निर्माण करना चाहते है कि केवल वही कॉलम प्राप्त करता है जो मैं चाहता हूं।

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 केवल एक कॉलम होता है। यदि यह एक महत्वपूर्ण अंतर है, तो मैं इसे "सही" कर सकता हूं, लेकिन आपको इस बारे में सोचना चाहिए कि उस मूल्य का उपयोग कैसे किया जाएगा और यदि इसके लिए विभिन्न संभावित संरचनाएं सुविधाजनक हैं।