Как воспроизвести этот пример Wolfram Solve в sympy

Question

Я хочу использовать sympy для воспроизведения результата, полученного на языке Wolfram Language.

Используя Wolframcloud, это выражение

Solve[m^2+m*n==500 && m>n,{m,n},PositiveIntegers]

Дает искомый результат:

{{m->20,n->5}}

Как я могу воспроизвести это с помощью sympy?

Я пробовал

import sympy as sp
m,n = sp.symbols('m n',integer=True)
sp.solve(m**2 + m*n - 500, m,n)

, что дает

[(m, -m + 500/m)]

, что правильно, но не особенно полезно .

Обратите внимание, что этот вопрос основан на задаче Project Euler 9.

Answer 1

Вы должны использовать диофантин для целочисленных решений:

In [10]: m, n = symbols('m, n')                                                                                                                

In [11]: sols = diophantine(m**2 + m*n - 500, (m, n))                                                                                          

In [12]: sols                                                                                                                                  
Out[12]: 
{(-500, 499), (-250, 248), (-125, 121), (-100, 95), (-50, 40), (-25, 5), (-20, -5), (-10, -40), (-5, -95), (-4, -121), (-2, -248), (-1, -499)
, (1, 499), (2, 248), (4, 121), (5, 95), (10, 40), (20, 5), (25, -5), (50, -40), (100, -95), (125, -121), (250, -248), (500, -499)}

Это дает решения для целых чисел m, n. Вы можете отфильтровать решения для тех, которые удовлетворяют вашим условиям:

In [13]: {(m, n) for m, n in sols if m > n and m > 0 and n > 0}                                                                                
Out[13]: {(20, 5)}