Skill: NumPy

Best practices for numerical computing with NumPy including arrays, broadcasting, and vectorization.

When to Use

Apply this skill when doing numerical computing with NumPy — arrays, broadcasting, linear algebra, random sampling.

Arrays

• Use explicit dtypes (np.float64, np.int32) when creating arrays.
• Prefer np.zeros, np.ones, np.empty, np.arange, np.linspace over list-based construction.
• Use structured arrays or separate arrays instead of object arrays.

Vectorization

• Replace Python loops with vectorized NumPy operations wherever possible.
• Use broadcasting rules to operate on arrays of different shapes without explicit expansion.
• Use np.where() for conditional element-wise operations.