Kondisi dalam List Comprehension di Python
Jika perlu menetapkan kondisi dalam comprehension, maka kondisi tersebut ditulis di sebelah kanan dari objek yang dapat diiterasi (daftar, rentang angka):
list = [expression for element in iter if condition]
Saat menghasilkan daftar, Anda dapat menetapkan
kondisi tambahan untuk elemen-elemennya.
Mari kita tulis kondisi di mana hanya elemen genap
dari rentang 1 hingga 10 yang akan dimasukkan ke dalam daftar:
lst = [i for i in range(1, 10) if i % 2 == 0]
print(lst)
Setelah kode dijalankan, daftar baru dengan elemen genap akan ditampilkan:
[2, 4, 6, 8]
Dengan menggunakan comprehension, buatlah daftar yang hanya berisi elemen ganjil:
[1, 3, 5, 7, 9]
Diberikan daftar:
lst = [-6, -3, -1, 0, 2, 4]
Dengan menggunakan comprehension, buatlah darinya daftar baru yang hanya berisi angka-angka positif, termasuk nol.