如何从csv文件中选择3列系数的数据

Question

我希望根据数据集中的行索引绘制2种不同情况的列数，最好通过Pandas.DataFrame：

第一种情况：列索引[2,5,8，...，n + 2]

第二种情况：最后480列或列索引[961-1439] picture

我尝试过以下索引：

import pandas as pd
import numpy as np
import matplotlib.pyplot as plt

dft = pd.read_csv("D:\Test.csv" , header=None)
dft.head()

id_set = dft[dft.index % 2 == 0].astype('int').values
A = dft[dft.index % 2 == 1].values
B = dft[dft.index % 2 == 2].values
C = dft[dft.index % 2 == 3].values
data = {'A': A[:,0], 'B': B[:,0], 'C': C[:,0]}
df = pd.DataFrame(data, columns=['A','B','C'], index = id_set[:,0])

#1st scenario
j=0
index=[]
for i in range(1439):
    if j==2:
        j=0
        continue
    else:
        index.append(i)
        j+=1
print(index)

#2nd scenario
last_480 = df.[0:480][::-1]

我找到了post1和post2，但事实并非如此！

如果有人可以帮助我，我将不胜感激。

Answer 1

第一种情况：

df.iloc[:, 2::3]

这里的切片表示从第二个开始的所有行，列，然后是第二个。

第二种情况：

df.iloc[:, :961:-1]

这里的切片表示从列表末尾到961的所有行，列。

编辑：

import matplotlib.pyplot as plt
import seaborn as sns

senario1 = df.iloc[:, 2::3].copy()
sns.lineplot(data = senario1.T)

您可以将切片的副本保存到另一个变量，然后由于要按行绘制图形，因此需要对切片的矩阵进行转置（这会使您的行变成列）。