Conditional statements inside a nested for loop

@harry_zhao

maybe you can reduce your codeblock to