如何为一个输入变量添加不同的金额?

时间:2016-01-21 02:39:15

标签: python variables main

我目前有一个嵌套列表,其中包含太阳的名称,最长和最短距离,以及这些行星距离01/01/16的时间范围(以周为单位)。变量是用户输入的周数,但是因为所有的行星最初都是在' -x'默认情况下,我有+/-用户时间所需的时间,以便正确定位它们。

Week=int(input("How many weeks would you like to see into the Solar System's future?   "))
TimeFormat=365.25*60*60*24

PlanetData = [
    ['Mercury', 69.8, 46.0, (Week+1.5)/52 * TimeFormat],
    ['Venus', 108.9, 107.5, (Week-9)/52 * TimeFormat],
    ['Earth', 152.1, 147.1, (Week-1.5)/52 * TimeFormat],
    ['Mars', 249.2, 206.7, (Week-21)/52 * TimeFormat],
    ["Halley's Comet",5250, 87.7, (Week+1.54e3)/52 * TimeFormat],
]

此刻此工作正常,但我正在尝试使用所有变量(包括用户"周"输入)创建一个主函数,然后将常量分开。这让我想到了如上所示定义行星而没有变量的问题。我不明白如何只有' + 1.5'在行星定义中,然后在功能部分单独添加用户输入。

下面给出了如何在函数中使用此列表的示例,但是大约有8个函数使用每个行星的不同信息组合。

def MapPlanet(Max, Min, Time):
    SCALE = 1e9
    theta, r = SolveOrbit(Max * SCALE, Min * SCALE, Time)
    x = -r * cos(theta) / SCALE
    y = r * sin(theta) / SCALE
    return x, y

def DrawPlanet(Name, Max, Min, Time):
    x, y = MapPlanet(Max, Min, Time)
    Planet = Circle((x, y), 8)
    plt.figure(0).add_subplot(111, aspect='equal').add_artist(Planet)
    plt.annotate(Name, xy=((x+5),y),color='red')

这将在主函数中执行,如下所示:

def Main():

    Week=int(input("How many weeks would you like to see into the Solar System's future?   "))

    for Name, Max, Min, Time in PlanetData:
        MapPlanet(Max, Min, Time)
        DrawPlanet(Name, Max, Min, Time)

1 个答案:

答案 0 :(得分:2)

在这种情况下你可以做的是定义一个函数,它将通过闭包存储每个行星的输入数据并计算出适当的值:

TimeFormat=365.25*60*60*24

def planet_time(data):
    def value(Week):
        return (Week+data)/52 * TimeFormat
    return value(Week)

然后,您可以使用此函数定义每个行星(下面的代码是代码的简化版本,但PlanetData已完成):

Week = 4

PlanetData = [
    ['Mercury', 69.8, 46.0, planet_time(1.5)],
    ['Venus', 108.9, 107.5, planet_time(-9.0)],
    ['Earth', 152.1, 147.1, planet_time(-1.5)],
    ['Mars', 249.2, 206.7, planet_time(21.0)],
    ["Halley's Comet",5250, 87.7, planet_time(1.54e3)],
]

for Name, Max, Min, Time in PlanetData:
    print("{}, {}".format(Name, Time))

此代码打印:

Mercury, 3337823.07692                                                                                                                                                                                                                                 
Venus, -3034384.61538                                                                                                                                                                                                                                  
Earth, 1517192.30769                                                                                                                                                                                                                                   
Mars, 15171923.0769                                                                                                                                                                                                                                    
Halley's Comet, 937017969.231