Merge pull request #2 from hernandanielg/master

Updated List Comprehensions and Iterables and Iterators

Author: arsho

DataTypes/ListComprehensions.py

@@ -10,13 +10,4 @@
 y=int(raw_input())
 z=int(raw_input())
 n=int(raw_input())
-final_ar=[]
-for i in range(0,x+1):
-    for j in range(0,y+1):
-        for k in range(0,z+1):
-            tmp=i+j+k
-            if(tmp!=n):
-                ar=[i,j,k]
-                final_ar.append(ar)
-                
-print final_ar
+print([ [i,j,k] for i in range(x+1) for j in range(y+1) for k in range(z+1) if i+j+k != n ])

Itertools/IterablesandIterators.py

@@ -5,16 +5,17 @@
 Author    : Ahmedur Rahman Shovon
 Created   : 15 July 2016
 '''
-import itertools
+from itertools import *
+
 n = int(input())
-s_ar = input().strip().split()
-s = ''.join(s_ar)
+s = str(input())
 k = int(input())
-combination = list(itertools.combinations(s,k))
-combination_count = len(combination)
-a_count = 0
 
-for single_combination in combination:
-    if 'a' in single_combination:
-        a_count += 1
-print(a_count / combination_count)
+pairs = list(combinations(s.split(),k))
+count = 0
+
+for pair in pairs:
+    if 'a' in pair:
+            count += 1
+
+print(count/len(pairs))