https://camera-optical-zoom.blogspot.com/2024/01/sklearn-truth-value-of-array-with-more.html