その計算を行うPython関数を書くことができますか？
もちろん！ここにPythonの関数がある：

```
import math

def ball_speed(h, g=9.8):
    v = math.sqrt(2 * g * h)
    return v
```

この関数は相対的な高さ `h` を入力として、重力による加速度 `g` （デフォルトでは `9.8` m/s^2 に設定されている）を受け取ります。この関数はインパクト直後のボールのスピードを返します。