python嵌套循环简写,如何简化python中的两个嵌套循环

I'm trying to simplify two neested for loops in python but I cant resolve this.

My code:

head = [[1, 2], [3, 4]]

temp = []

for array in head:

for element in array:

temp.append(element)

print(temp)

========OUTPUT========

[1, 2, 3, 4]

I try:

head = [[1, 2], [3, 4]]

temp = []

for array in head:

temp += [element for element in array]

print(temp)

But only can simplify one loop

EDIT: SOLUTION

Specific solution for my case by @serafeim:

head = [[1, 2], [3, 4]]

print([element for array in head for element in array])

Other solutions:

By anon

from functools import reduce

head = [[1, 2], [3, 4]]

print(reduce(list.__add__, head))

By: @chepner

from itertools import chain

head = [[1, 2], [3, 4]]

print([x for x in chain.from_iterable(head)])

By: @R-zu

import numpy as np

head = [[1, 2], [3, 4]]

print(np.array(head).reshape(-1).tolist())

解决方案

This is already available from the itertools module.

from itertools import chain

temp = [x for x in chain.from_iterable(head)]

# or just temp = list(chain.from_iterable(head))