Defining list from a function with multiple arguments

Question

I have the following function:

def func(x, y, z):
  d = -5.9
  T = 230.0

  f = x - y * T + z + d

  return f

From this, I want to make a single list of f-values extracted from func(x, y, z). The values x, y, and z are three different lists of values.

My question is, how do I make the list of f-values? I have tried the following, but it did not work properly:

f_list = [func(x, y, z) for x in x_list for y in y_list for z in z_list]

Netwave · Accepted Answer · 2018-05-12 09:30:09Z

3

I think you want to use zip:

f_list = [func(*vals) for vals in zip(x_list, y_list, z_list)]

or for clearance:

f_list = [func(x, y, z) for x, y, z in zip(x_list, y_list, z_list)]

import itertools
f_list  = list(itertools.starmap(func, zip(x_list, y_list, z_list)))

answered May 12, 2018 at 9:21

Netwave

43.2k6 gold badges69 silver badges110 bronze badges

Sign up to request clarification or add additional context in comments.

1 Answer 1