To make it for columns instead of rows in Stefan’s second example, just use [:,i]. Also, use $ before variables with @btime to get proper timings.
[:,i]
$
@btime