Can you do this as some kind of numpy splat, rather than a list comprehension?