Python: Danh sách hiểu để gán các giá trị khác nhau

Question

Tôi đang tạo danh sách 2D và tôi muốn khởi tạo danh sách đó bằng cách hiểu danh sách. Tôi muốn nó để làm một cái gì đó như thế này:

[[x for i in range(3) if j <= 1: x=1 else x=2] for j in range(3)] 

nên nó phải trả lại một cái gì đó như:

[[1,1,1], 
[1,1,1], 
[2,2,2]] 

Làm thế nào tôi có thể đi về việc này?

Cảm ơn sự giúp đỡ của bạn.

Answer 1

Có vẻ như bạn đang tìm kiếm một cái gì đó như thế này:

[[1 if j <= 1 else 2 for i in range(3)] for j in range(3)] 

Các Python conditional expression là một chút khác biệt so với những gì bạn có thể được sử dụng để nếu bạn đang đến từ một cái gì đó như C hoặc Java:

Khái niệm x if C else y đánh giá lại đầu tiên C (không x); nếu C là đúng, x được đánh giá và giá trị của nó được trả về; nếu không, y được đánh giá và giá trị của nó được trả về.

Một cách hơi ngắn hơn để làm điều tương tự là:

[[1 if j <= 1 else 2]*3 for j in range(3)] 

Answer 2

phản ứng của Greg là đúng, mặc dù một biểu thức đơn giản hơn nhiều và nhanh hơn để tạo ra kết quả mong muốn của bạn sẽ

[[j] * 3 for j in (1, 1, 2)] 

tức là, hãy nhớ rằng for không cần phải áp dụng cho một chỉ range ;-), danh sách nhân, tồn tại, và như vậy ;-).

Answer 3

Hãy thử điều đó

>>> [[(1 if j<1 else 2) for i in range(3)] for j in range(3)] 
[[1, 1, 1], [2, 2, 2], [2, 2, 2]] 

Lần thứ hai j = 1 nên j < 1 thất bại