Skip to content

Commit c39488e

Browse files
committed
use ternary operators and sum comprehension
1 parent d2bac19 commit c39488e

File tree

1 file changed

+5
-7
lines changed

1 file changed

+5
-7
lines changed

SeamCarverUtilities.py

Lines changed: 5 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -4,15 +4,13 @@ def printVerticalSeam(sc):
44
totalSeamEnergy = 0
55
for row in range(sc.height()):
66
for col in range(sc.width()):
7-
seam_col = seam[row_index]
8-
lmarker = ' '
9-
rmarker = ' '
10-
if col == seam_col:
11-
lmarker = '['
12-
rmarker = ']'
13-
totalSeamEnergy += sc.energy(col, row)
7+
seam_col = seam[row]
8+
lmarker = '[' if col == seam_col else ' '
9+
rmarker = ']' if col == seam_col else ' '
1410
print '{:s}{:>6d}{:s}'.format(lmarker, sc.energy(col, row), rmarker),
1511
print
12+
13+
totalSeamEnergy = sum(sc.energy(col, row) for col, row in zip(seam, range(sc.height())))
1614
print "\nTotal seam energy: {:d}".format(totalSeamEnergy)
1715

1816

0 commit comments

Comments
 (0)