Python: if-else внутри Генератора Списков

Долго искал в интернете, на зная как обозначить данную проблему. В итоге, гуглить стоит что-то наподобие:

“Creating a Python list comprehension with an if”  “Python: if/else in list comprehension?”

Задачка: создать на Питоне список от 1 до 100, где каждый нечетный элемент списка взят в квадрат.

Как это бы выглядело на Си:

1. int main(int argc, char **argv)
2. {
3.   int i;
4.   int mass[100];
6.   for (i = 1; i < 101; ++i) {
7.       mass[i-1] = (i % 2 == 1) ? (i*i): i;
8.   }
10.   printf("mass[] = [ ");
11.   for (i = 0; i < 100; ++i) {
12.       printf("%d ,", mass[i]);
13.   }
14.   printf(" ]n");
15. }

А на питоне при помощи генераторов списков всего две строчки:

1. list = [ x ** 2 if x % 2 else x for x in range (1, 101)]
2. print list

Для более сложных ветвистых условий, видимо, можно определить метод типа функтора:

1. def foo(arg):
2.   if arg % 2 == 1:
3.       return arg **2
4.   else:
5.       return arg
7. list = [ foo(i) for i in range(1, 101) ]
8. print list

Я уверен есть еще несколько способов решения этой задачки и методов использования генераторов списков. А вывод этой задачки таков:

$ python test1.py

[1, 2, 9, 4, 25, 6, 49, 8, 81, 10, 121, 12, 169, 14, 225, 16, 289, 18, 361, 20, 441, 22, 529, 24, 625, 26, 729, 28, 841, 30, 961, 32, 1089, 34, 1225, 36, 1369, 38, 1521, 40, 1681, 42, 1849, 44, 2025, 46, 2209, 48, 2401, 50, 2601, 52, 2809, 54, 3025, 56, 3249, 58, 3481, 60, 3721, 62, 3969, 64, 4225, 66, 4489, 68, 4761, 70, 5041, 72, 5329, 74, 5625, 76, 5929, 78, 6241, 80, 6561, 82, 6889, 84, 7225, 86, 7569, 88, 7921, 90, 8281, 92, 8649, 94, 9025, 96, 9409, 98, 9801, 100]