在python中自动计算旋转周期

Question

我是编程初学者，我使用python。我有一个计算恒星自转周期的代码，但是每次我都要更改恒星ID，这需要我付出努力和时间来完成。 我能自动更改星号吗？你知道吗

from lightkurve import search_lightcurvefile
lcf = search_lightcurvefile('201691589').download()  ## star Id = 201691589
lc = lcf.PDCSAP_FLUX.remove_nans()
pg = lc.to_periodogram()
Prot= pg.frequency_at_max_power**-1
print Prot

我将所有要使用的“stars\u ID”保存在一个txt文件中(starID.txt文件)有10000行，我想自动计算旋转周期（Prot），这样代码就可以从txt文件中一个接一个地获取星ID并进行计算，然后将星ID和Prot保存在csv文件中（两列：'星ID'，'Prot'）。你能帮我做吗。你知道吗

Answer 1

这应该让你接近，但我没有一堆明星身份证，也不是在我的领域。你知道吗

要点：

1. 使用^{}模块读写文件。你知道吗
2. 当您有需要多次调用的代码时（对于逻辑分组，通常甚至只调用一次），您需要考虑将其打包成function

你还有其他的研究方向。如果我不尝试使事情比基本循环更简洁一点，那么代码将相当长，我尽量不使它太简洁。希望你能继续跟进。你知道吗

from lightkurve import search_lightcurvefile

import csv

# You need to read the file and get the star IDs. I'm taking a guess here that
# the file has a single column of IDs
with open('starID.txt') as infile:
    reader = csv.reader(infile)
    # Below is where my guess matters. A "list comprehension" assuming a single 
    # column so I just take the first value of each row.
    star_ids = [item[0] for item in reader] 


def data_reader(star_id):
    """
    Function to read periodogram for a star_id

    Returns [star_id, periodogram]
    """

    lcf = search_lightcurvefile('201691589').download()
    lc = lcf.PDCSAP_FLUX.remove_nans()
    pg = lc.to_periodogram()
    Prot= pg.frequency_at_max_power**-1

    return [star_id, Prot]


# Now start calling the function on your list of star IDs and storing the result
results = [] 
for id_number in star_ids:
    individual_result = data_reader(id_number) # Call function
    results.append(individual_result) # Add function response to the result collection


# Now write the data out
with open('star_results.csv', 'w', newline='') as outfile:
    writer = csv.writer(outfile)
    writer.writerows(results)