Làm thế nào để vẽ một âm mưu scree trong python? [đóng cửa]

Tôi đang sử dụng phân rã vectơ số ít trên một ma trận và thu được ma trận U, S và Vt. Tại thời điểm này, tôi đang cố gắng chọn ngưỡng cho số lượng kích thước được giữ lại. Tôi đã được đề nghị để xem xét một âm mưu scree nhưng đang tự hỏi làm thế nào để đi về âm mưu nó trong numpy. Hiện tại, tôi đang thực hiện các thao tác sau bằng thư viện numpy và scipy trong python:

U, S, Vt = svd(A)

Bất kỳ đề xuất?

Dưới đây là một ví dụ có thể được dán vào dấu nhắc IPython và tạo một hình ảnh như bên dưới (nó sử dụng dữ liệu ngẫu nhiên):

import numpy as np
import matplotlib
import matplotlib.pyplot as plt

#Make a random array and then make it positive-definite
num_vars = 6
num_obs = 9
A = np.random.randn(num_obs, num_vars)
A = np.asmatrix(A.T) * np.asmatrix(A)
U, S, V = np.linalg.svd(A) 
eigvals = S**2 / np.sum(S**2)  # NOTE (@amoeba): These are not PCA eigenvalues. 
                               # This question is about SVD.

fig = plt.figure(figsize=(8,5))
sing_vals = np.arange(num_vars) + 1
plt.plot(sing_vals, eigvals, 'ro-', linewidth=2)
plt.title('Scree Plot')
plt.xlabel('Principal Component')
plt.ylabel('Eigenvalue')
#I don't like the default legend so I typically make mine like below, e.g.
#with smaller fonts and a bit transparent so I do not cover up data, and make
#it moveable by the viewer in case upper-right is a bad place for it 
leg = plt.legend(['Eigenvalues from SVD'], loc='best', borderpad=0.3, 
                 shadow=False, prop=matplotlib.font_manager.FontProperties(size='small'),
                 markerscale=0.4)
leg.get_frame().set_alpha(0.4)
leg.draggable(state=True)
plt.show()