如何在python中创建用户图，以便在两个用户之间的距离小于某个值时这两个用户之间存在边？

import networkx as nx
import matplotlib.pyplot as plt

G = nx.Graph()

users = ["user1", "user2", "user3", "user4"]
G.add_nodes_from(users)

distance_threshold = 5

for i in range(len(users)):
    for j in range(i+1, len(users)):
        # 计算两个用户之间的距离
        distance = calculate_distance(users[i], users[j])
        
        # 如果距离小于阈值，则添加边
        if distance < distance_threshold:
            G.add_edge(users[i], users[j])

nx.draw(G, with_labels=True)
plt.show()