如何使用CSV列标题作为情节标签

Question

我正在尝试使用三个列标题（“ node_pot_gauche_i”，“ node_lisse_center_i”，“ node_pot_droite_i”）作为标签，同时在我的后制作Python脚本中绘制ANSYS APDL分析结果。

我目前要做的是，当我在循环中绘制时，我会手动添加标签：

# Parcours de tous les fichiers csv
            for i in range(len(model[idx]['asb_path'])):

                # Parcours a l'envers des indices pour positionner les graphs
                j = len(model[idx]['asb_path'])-i-1

                # Lecture du fichier csv possédant des headers
                potiso = pd.read_csv(model[idx]['potiso_path'][i], index_col=0)
                asb = pd.read_csv(model[idx]['asb_path'][i], index_col=0)

                # Plot des résultats sur un subplot de la figure créer en haut
                axs[j].plot(potiso, label='NODE_POT_ISO')
                axs[j].plot(asb['NODE_POT_GAUCHE_'+str(i+1)], label='NODE_POT_GAUCHE_'+str(i+1))
                axs[j].plot(asb['NODE_POT_DROITE_'+str(i+1)], label='NODE_POT_DROITE_'+str(i+1))
                axs[j].set_title(os.path.basename(model[idx]['asb_path'][i]))
                axs[j].set_xlabel('time(s)')
                axs[j].set_ylabel('displacement(m)')
                axs[j].legend()
                axs[j].grid(True)

            # Enregistrement au format .png
            plt.savefig(fig_path)

我想要的是自动使用数据框中的列标题，并在绘图时自动使用它们作为标签更改CSV组织，而不会打扰我的Python后过程脚本。

我似乎找不到我的问题的答案，我尝试了不同的事情，例如让它空白：

# Parcours de tous les fichiers csv
            for i in range(len(model[idx]['asb_path'])):

                # Parcours a l'envers des indices pour positionner les graphs
                j = len(model[idx]['asb_path'])-i-1
                # Lecture du fichier csv possédant des headers
                asb = pd.read_csv(model[idx]['asb_path'][i], index_col=0)

                # Plot des résultats sur un subplot de la figure créer en haut
                axs[j].plot(asb)
                axs[j].set_title(os.path.basename(model[idx]['asb_path'][i]))
                axs[j].set_xlabel('time(s)')
                axs[j].set_ylabel('displacement(m)')
                axs[j].legend()
                axs[j].grid(True)

            plt.savefig(fig_path)

但是我遇到了以下错误：

没有发现传奇的唱片公司的艺术家。请注意，标签以下划线开头的艺术家在没有参数的情况下被忽略。

必须有一种方法来提取列的名称，然后将它们作为我的情节标签输入输入。 在此先感谢您的任何帮助！

Answer 1

有人在其他地方回答了我，这使我在这里分享答案：

# Parcours de tous les fichiers csv
            for i in range(len(model[idx]['asb_path'])):

                # Parcours a l'envers des indices pour positionner les graphs
                j = len(model[idx]['asb_path'])-i-1
                # Lecture du fichier csv possédant des headers
                asb = pd.read_csv(model[idx]['asb_path'][i], index_col=0)

                # Plot des résultats sur un subplot de la figure créer en haut
                axs[j].plot(asb, label = list(asb.columns))
                axs[j].set_title(os.path.basename(model[idx]['asb_path'][i]))
                axs[j].set_xlabel('time(s)')
                axs[j].set_ylabel('displacement(m)')
                axs[j].legend()
                axs[j].grid(True)

            plt.savefig(fig_path)

重要的部分是：

axs[j].plot(asb, label = list(asb.columns))

或：

axs[j].plot(asb, label = asb.columns)

它基本上使用了数据框架对象的列属性。您还可以循环浏览每一列并自己构建列表。 （在这里提供更多信息： https：// www。 geeksforgeeks.org/how-to-get-column-names-names-in-pandas-dataframe/ ）

i由于我不是在使用asb.columns而无法使用它作为列表，这就是为什么它以前没有工作的原因。

---

编辑：小心，如果只有一列是绘制：

axs[j].plot(asb, label = list(asb.columns))

将返回['header_name']。

和：

axs[j].plot(asb, label = asb.columns)

将返回索引。

因此，如果只有一列，则应使用：

axs[j].plot(asb, label = asb.columns[0])