在numpy距离对称矩阵中获取对角正方形坐标

Question

如何获取对角线上的大矩形的坐标。

例如黄色 [0,615]、[615,1438]、[1438,1526]

import numpy as np; 
import pandas as pd
from sklearn.metrics.pairwise import cosine_similarity

df = pd.DataFrame(array) # array is image numpy
df.shape #(1526, 360)
s = cosine_similarity(df) #(1526, 1526)
plt.matshow(s)

我尝试在第一行获取峰值，但有噪声信息

speak = 1-s[0]

peaks, _ = find_peaks(speak, distance=160, height=0.1)
print(peaks, len(peaks))
np.diff(peaks)

plt.plot(speak)
plt.plot(peaks, speak[peaks], "x")
plt.show()

更新，添加另一个例子 并上传到colab完整脚本https://colab.research.google.com/drive/1hyDIDs-QjLjD2mVIX4nNOXOcvCZY4O2c?usp=sharing

Answer 1

使用DBSCAN 聚类解决（在同一问题中找到DBSCAN for clustering of geographic location data）

from sklearn.cluster import DBSCAN

clustering = DBSCAN(eps=.5, min_samples=10).fit_predict(s)

peaks = np.where(clustering[:-1] != clustering[1:])[0]

Answer 2

cosine_similarity 的所有对角元素都是相同的。所以你应该寻找附近值的变化。

你可以试试这个：

factor = 1.01
look_nearby = 1

changes = []
for i in range(look_nearby, s.shape[0]-look_nearby):
    if s[i, i+look_nearby] > factor*s[i, i-look_nearby] or factor*s[i, i+look_nearby] < s[i, i-look_nearby]:
        changes.append(i)
        
print(changes)

根据您的偏好设置factor 值（因为您不希望在第一张图像的输出中出现 (1200, 1200)）并根据 s 的值。

Answer 3

使用np.diag(df) 获取对角元素列表。如果屏幕截图中的颜色代表低于/高于某个值（可能为零），请检查值何时超过阈值。