次に、このコードを代わりに関数に変換します。機能が何をするか、その入力、出力の説明を含む、numpy形式でドキュストリングを追加します。
Now, convert this code into a function instead. Add a docstring in numpy format to it containing a description of what the function does, its inputs and outputs.