Have you ever felt that you are not coding Python as productive as you want to be? Have you ever found yourself repeating the same statements consecutively but each with slight changes?
Well then, is guide is for you. The beauty of Python is that it gives us several valuable constructs to avoid repetition and to do more work with less code.
You may already know such gems as what I call the "two-variable swap trick" involving the use of tuples or the "multiple value input” trick that gets many variables from
input()
by running the resulting line through split(" ")
, or even the one-liner fit reversing an array using [::-1]
, but this article will give you even more shortcuts to use.Variable numbers of arguments
You might know that there are some C functions such as
printf()
, which take a variable number of arguments depending on the value of the first one. This is easy to replicate in Python by using list expansion after the initial positional arguments.def fun1(*a):
res=1
for ele in a:
res*=ele
return res
ans=fun1(5,6,9,6,9,6,6)
print(ans) # prints `524880`
(from 10 Python Tricks That Will Wow You)
Here's another more organic example, a calculator that makes use of positional arguments:
OP_ADD = 1
OP_SUBTRACT = 2
OP_MULTIPLY = 3
#...
def calculator(op, *var):
if op == OP_ADD:
a, b = *var
return a + b
elif op == OP_NEG:
a = *var
return -a
#...
Selectively printing text
Have you ever found yourself writing code like this?
if foo == True:
print("All is well!")
else:
print("We may have a problem")
You don't have to do that! This can be done in one line using the condensed form of the
if
statement.print("All is well!" if foo == True else "We may have a problem")
Make a table of element occurrences
By making use of the
count()
method as a key, we can convert arrays to dictionaries to find how many times each element occurs.Two-way comparisons
In Python, you can chain two less-than or greater-than operators to test whether a number is within a range. No more having to test them separately!
if 1 < x < 10:
#...
Enhanced tracing using icecream
icecream
It's happened to most of us. The debugger goes kaput, and it forces us to sprinkle
print()
statements everywhere to trace program execution while finding a bug. Now seeing is everything, and most people can't write good words that help someone identify where control flow is (I know I can’t).That's where
icecream
comes in handy. icecream
is a PyPI module that contains a function you can place anywhere in the program, and when you call it without arguments, it will tell you the file and line number, as well as function it's in as well as the current time.When you do call it with an argument, however, it will print the name and value of the variable you passed. It's convenient to avoid mass repetition from writing a bunch of your own
print()
statements.from icecream import ic
def hello(user:bool):
if user:
ic()
else:
ic()
hello(user=True)
num1 = 30
num2 = 40
ic(num1)
ic(num2)
(from Stop Using Print to Debug in Python. Use Icecream Instead)
Antipatterns
Having said all that, make sure you avoid the following pitfalls while coding:
- Using list comprehensions on a vast array. This will use unnecessarily large amounts of memory.
- Using
for the iterator inrange(len(<list>))
loops. This forces you to get the list length again usingfor
. It is better to uselen()
in such scenarios to get both the value and index. Additionally, avoid usingenumerate()
syntax if you need indices in the loop.for i in ...
- Using square brackets
to access dictionary keys if they may not exist. Instead of handling[]
exceptions (and it's tempting to omitKeyError
groups when accessing random dictionary keys), use thetry
method of dictionaries, which will returnget()
if the key does not exist. The resultingNone
conditional is much easier to handle in code.if
So as you just saw, Python is a very rich language that gives you more straightforward ways of programming the same thing. And where the functionality is not built-in, there is usually a PyPI module that does it.