https://keroseneheaterreview.blogspot.com/2023/11/sklearn-truth-value-of-array-with-more.html