Das deutsche Python-Forum

Code: Alles auswählen

for f in funcs:
      for ... # deine Schleife

Code: Alles auswählen

In [18]: func = [min, max]

In [19]: a1 = [1, 2, 3]

In [20]: a2 = [5, 6, 7, 8]

In [21]: for a, b, f in zip(a1, a2, func):
    ...:     print(f(a, b))
    ...:     
1
6

Code: Alles auswählen

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
Created on Wed May 23 14:33:29 2018

@author: Rainer
"""

import numpy as np

def f1(y):
    r = y**2
    return r

def f2(y):
    r = y*2
    return r

def f3(y):
    r = y*3
    return r

def main():
    
    a1 = np.array([[1,2,3],[2,3,4],[2,2,2]])
    a2 = np.array([[1,2,3],[2,3,4],[2,2,2]])
    
    i = 0
    for a, b in zip(a1, a2):
        y = sum(a + b)
        print(y)
        if i == 0:
            x = f1(y)
            print(i, x)
        if i == 1:
            x = f2(y)
            print(i, x)
        if i == 2:
            x = f3(y)
            print(i, x)
#        x = funcs[i](y)
        
        i += 1
        
    # was ich gern hätte:
        
    funcs = [[f1], [f2], [f3]]
#    funcs = np.array(funcs)
    print(funcs)
    print(type(funcs))
    for a, b, f in zip(a1, a2, funcs):
        y = sum(a + b)
        x = f(y)
        print(x)
        
if __name__ == '__main__':
    main()